Update app.py

app.py

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