Update app.py

app.py

@@ -112,15 +112,15 @@ def generate_text(prompt, system_prompt):
         {"role": "user", "content": prompt}
       ]
     try:
-        response = client.workers.ai.with_streaming_response.run(
+        result = client.workers.ai.run(
             account_id=ACCOUNT_ID,
             model=CHAT_MODEL,
             messages=messages,
         )
 
-        print(f'response: {response}')
+        print(f'response: {result["response"]}')
 
-        return response
+        return result["response"]
         
     except Exception as e:
         raise gr.Error(str(e))
@@ -128,7 +128,7 @@ def generate_text(prompt, system_prompt):
         
 def gen(promtp: str, system_prompt: str):
     text = generate_text(promtp, system_prompt)
-    image = generate_image(text)
+    image = generate_image(str(text))
     return text, image
 
 # Gradio Interface