added kosmos2 api

app.py

@@ -2,21 +2,29 @@ import gradio as gr
 from gradio_client import Client
 
 #fusecap_client = Client("https://noamrot-fusecap-image-captioning.hf.space/")
-fuyu_client = Client("https://adept-fuyu-8b-demo.hf.space/")
+#fuyu_client = Client("https://adept-fuyu-8b-demo.hf.space/")
+kosmos2_client = Client("https://ydshieh-kosmos-2.hf.space/")
 
 def get_caption(image_in):
-    
+    """
     fuyu_result = fuyu_client.predict(
 	    image_in,	# str representing input in 'raw_image' Image component
 	    True,	# bool  in 'Enable detailed captioning' Checkbox component
 		fn_index=2
     )
+    """
+
+    kosmos2_result = kosmos2_client.predict(
+        image_in,	# str (filepath or URL to image) in 'Test Image' Image component
+        "Brief",	# str in 'Description Type' Radio component
+        fn_index=4
+    )
 
     # Find the last occurrence of "."
-    last_period_index = fuyu_result.rfind('.')
+    last_period_index = kosmos2_result.rfind('.')
 
     # Truncate the string up to the last period
-    truncated_caption = fuyu_result[:last_period_index + 1]
+    truncated_caption = kosmos2_result[:last_period_index + 1]
 
     # print(truncated_caption)
     print(f"\n—\nIMAGE CAPTION: {truncated_caption}")