Update app.py

app.py

@@ -1,16 +1,29 @@
 import gradio as gr
 from gradio_client import Client
+import json
 
 def get_caption(image_in):
     client = Client("https://vikhyatk-moondream1.hf.space/")
     result = client.predict(
-        image_in,  
-        "Describe the image", 
+        image_in,  # filepath in 'image' Image component
+        "Describe the image",  # str in 'Question' Textbox component
         api_name="/answer_question"
     )
-    caption = result['choices'][0]['text']
+
+    # JSON 형태로 파싱
+    try:
+        result_json = json.loads(result)
+        caption = result_json['choices'][0]['text']
+    except json.JSONDecodeError:
+        print("Error: Response is not valid JSON.")
+        return ""
+    except (KeyError, IndexError):
+        print("Error: Invalid format in JSON response.")
+        return ""
+
     return caption
 
+
 def get_lcm(prompt):
     client = Client("https://latent-consistency-lcm-lora-for-sdxl.hf.space/")
     images = []