Update app.py

app.py

@@ -2,20 +2,16 @@ import gradio as gr
 from gradio_client import Client
 
 def get_caption(image_in):
-    # Correct indentation before client initialization
     client = Client("https://vikhyatk-moondream1.hf.space/")
-    
-    # Align the call to 'client.predict' with the line above
-    # Also, the parameter 'prompt' is not defined in 'get_caption'. 
-    # It should be 'image_in' to match the function signature.
     results = client.predict(
-        image_in,  # Use the 'image_in' parameter as input to the model
-        "Describe the image",  # Assuming this is a constant parameter for the model
+        image_in,  # 'image_in'은 이미지 파일의 경로
+        "Describe the image",  # 'Describe the image'는 모델에 제출되는 고정 쿼리
         api_name="/answer_question"
     )
-    # Print and return the results assuming it contains the description
-    print(results)
-    return results
+    # 결과 튜플에서 설명 텍스트만 추출하여 반환
+    caption = results['data'][0]['text']  # API 응답의 구조에 따라 수정될 수 있음
+    print(caption)
+    return caption
 
 def get_lcm(prompt):
     client = Client("https://latent-consistency-lcm-lora-for-sdxl.hf.space/")