Update app.py

app.py

@@ -33,7 +33,7 @@ def respond(message, history):
 
     # プロンプトの作成
     prompt = pd.DataFrame(
-        {"prompt": [message], "num_inference_steps": 25}
+        {"query": [message]}
     )
 
     print(prompt)
@@ -46,37 +46,13 @@ def respond(message, history):
         # モデルサービングエンドポイントに問い合わせ
         response = requests.request(method="POST", headers=headers, url=local_endpoint, data=data_json)    
         response_data = response.json()
-        #print(response_data["predictions"])
-
-        # numpy arrayに変換
-        im_array = np.array(response_data["predictions"], dtype=np.uint8)
-        #print(im_array)
-        # 画像に変換
-        im = Image.fromarray(im_array, 'RGB')
-
-    
-        # debug
-        #image_url = "https://upload.wikimedia.org/wikipedia/commons/thumb/e/ec/Mona_Lisa,_by_Leonardo_da_Vinci,_from_C2RMF_retouched.jpg/687px-Mona_Lisa,_by_Leonardo_da_Vinci,_from_C2RMF_retouched.jpg"
-        #print("image_url:", image_url)
-        #im = Image.open(io.BytesIO(requests.get(image_url).content))
-        #numpydata = np.asarray(im)
-
-        rawBytes = io.BytesIO()
-        im.save(rawBytes, "PNG")
-        rawBytes.seek(0)  # ファイルの先頭に移動
-        # base64にエンコード
-        image_encoded = base64.b64encode(rawBytes.read()).decode('ascii')
-        #print(image_encoded)
-
-        # マークダウンに埋め込み
-        embed_image_markdown = f"![](data:image/png;base64,{image_encoded})"
-        #print(embed_image_markdown)
+        print(response_data["predictions"])
     
     except Exception as error:
         response_data = f"ERROR status_code: {type(error).__name__}" 
         #+ str(response.status_code) + " response:" + response.text    
 
-    return embed_image_markdown
+    return response_data["predictions"]
 
 
 theme = gr.themes.Soft(