Update app.py

app.py

@@ -37,6 +37,14 @@ def predict(image_input, question):
     
     # 修正：使用 text 參數而不是直接傳遞 messages
     result = pipe(text=messages)
+    
+    # 修正輸出格式：提取 assistant 的回應
+    if isinstance(result, list):
+        for message in result:
+            if isinstance(message, dict) and message.get("role") == "assistant":
+                return message.get("content", "")
+    
+    # 備用方案
     return result[0]["generated_text"]
 
 # Gradio 介面（同時支援 UI 和 API）
@@ -49,5 +57,5 @@ iface = gr.Interface(
 )
 
 # 啟動應用程式
-if __name__ == "__main__":
+if __name__ == "__main__":  # 修正語法錯誤：__name__ 不是 **name**
     iface.launch()