Update app.py

app.py

@@ -1,7 +1,7 @@
 import os
 import base64
 import gradio as gr
-from gradio_client import Client
+from gradio_client import Client, file
 import json
 
 MODEL_NAME = "QWEN"
@@ -50,16 +50,9 @@ def process_text(text_input, unit):
         system=f"You are a experienced {unit} doctor AI assistant." + SYS_PROMPT,
         api_name="/model_chat"
     )
-    print(job)
-    response = job.result()
+    response = job.outputs
     print(response)
-    return response[1][0]
-
-def encode_image_to_base64(image_input):
-    buffered = io.BytesIO()
-    image_input.save(buffered, format="JPEG")
-    img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
-    return img_str
+    return response[1][0][0]
 
 def process_image(image_input, unit):
     if image_input is not None:
@@ -67,21 +60,19 @@ def process_image(image_input, unit):
         #with open(image_input, "rb") as f:
         #   base64_image = base64.b64encode(f.read()).decode("utf-8")
         client = Client(client_vl)
-        # base64_image = encode_image_to_base64(image_input)
         prompt = f" You are a experienced {unit} doctor AI assistant." + SYS_PROMPT + "Help me understand what is in this picture and analysis."
-        client.predict(
-            "null",
-            image_input, 
+        res = client.predict(
+            file = file(image_input), 
             fn_index=5
         )
+        print(res)
         response = client.predict(
-            "null",
-            prompt,
+            text = prompt,
             fn_index=0
         )
         print(response)
-
-
+        return response[1][0][0]
+    
 def main(text_input="", image_input=None, unit=""):
     if text_input and image_input is None:
         return process_text(text_input,unit)