Update app.py

app.py

@@ -180,11 +180,20 @@ async def main(link, peoples="双人"):
         system_prompt = single_prompt;
     
     messages = system_prompt + "\n\n\n" + prompt
+
     
-    completion = client.predict(
-        user_message=messages,
+    job = client.submit(
+        messages,
+        "",
+        None,
+        [],
         fn_index=0
     )
+    while job.done:
+        completion = client.predict(
+            None,
+            fn_index=4
+        )
 
     print(completion)