Update app.py

app.py

@@ -46,7 +46,7 @@ def generate_image(prompt, aspect_ratio, realism):
                     glif_token_id = (glif_token_id+1)%no_of_accounts
                     glif_tokens_tried+=1
                     GLIF_API_TOKEN = os.getenv(f"GLIF_API_TOKEN{glif_token_id}")
-                    response_data = generate_image_async(prompt, aspect_ratio, realism)
+                    response_data = generate_image(prompt, aspect_ratio, realism)
                     glif_tokens_tried = 0
                     return response_data
                 response_data = "No credits available"
@@ -54,7 +54,7 @@ def generate_image(prompt, aspect_ratio, realism):
         elif "output" in response_data:
             url = response_data['output']
             image  = get_image_from_url(url)
-            return f"prompt:{response_data['inputs']['Prompt']}\noutput:{response_data['output']}"
+            return image
         else:
             return "Error: Unexpected response from server"
     except Exception as e: