Update app.py

app.py

@@ -40,8 +40,6 @@ def respond(message, history):
     ds_dict = {"dataframe_split": prompt.to_dict(orient="split")}
     data_json = json.dumps(ds_dict, allow_nan=True)
 
-    embed_image_markdown = ""
-
     try:
         # モデルサービングエンドポイントに問い合わせ
         response = requests.request(method="POST", headers=headers, url=local_endpoint, data=data_json)    
@@ -52,7 +50,7 @@ def respond(message, history):
         response_data = f"ERROR status_code: {type(error).__name__}" 
         #+ str(response.status_code) + " response:" + response.text    
 
-    return response_data["predictions"]
+    return response_data["predictions"][0]
 
 
 theme = gr.themes.Soft(