Update app.py

app.py

@@ -1,17 +1,20 @@
 import gradio as gr
+import os
 import spaces
 import json
 import re
-from gradio_client import Client
+from gradio_client import Client, handle_file
+
+kosmos2_token = os.environ.get("KOSMOS2_TOKEN")
 
 
 def get_caption_from_kosmos(image_in):
-    kosmos2_client = Client("https://ydshieh-kosmos-2.hf.space/")
+    kosmos2_client = Client("fffiloni/Kosmos-2-API", hf_token=kosmos2_token)
 
     kosmos2_result = kosmos2_client.predict(
-        image_in,	# str (filepath or URL to image) in 'Test Image' Image component
-        "Detailed",	# str in 'Description Type' Radio component
-        fn_index=4
+        image_input=handle_file(image_in),
+		text_input="Detailed",
+		api_name="/generate_predictions"
     )
 
     print(f"KOSMOS2 RETURNS: {kosmos2_result}")