Update app.py

app.py

@@ -3,7 +3,6 @@ import base64
 import gradio as gr
 from gradio_client import Client, file
 import time
-import json
 from gradio import utils
 
 MODEL_NAME = "QWEN"
@@ -16,8 +15,6 @@ def read(filename):
     
 SYS_PROMPT = read('system_prompt.txt')
 
-ABS_URL = "https://orrinin-hcare.hf.space/file="
-
 DESCRIPTION = '''
 <div>
 <h1 style="text-align: center;">家庭医生demo</h1>
@@ -57,7 +54,7 @@ def process_text(text_input, unit):
         time.sleep(0.1)
     response = job.result()
     print(response)
-    return response[-1][1][0][1]
+    return response[1][0][1]
 
 def process_image(image_input, unit):
     if image_input is not None:
@@ -76,20 +73,21 @@ def process_image(image_input, unit):
         )
         print(res5)
         
-        with open(res5, 'r') as f:
-            data = json.load(f)
-        data[0][0]['name'] = ABS_URL + image
-        with open(res5, 'w') as f:
-            json.dump(data, f, indent=4)
-            
-        job = client.submit(
+        res0 = client.predict(
             res5,
             prompt,
             fn_index=0
         )
+        
+        print(res0)
+        
+        response = client.submit(
+            fn_index=1         
+        )
         response = job.result()
         print(response)
-        return response[-1][0][0]
+        
+        return response[1][0][1]
     
 def main(text_input="", image_input=None, unit=""):
     if text_input and image_input is None: