Update app.py

app.py

@@ -148,10 +148,10 @@ def generate_image(prompt: str) -> str:
     try:
         client = Client()
         response = client.images.generate(
-            model="flux-schnell-black-forest-labs",
+            model="flux-pro",
             prompt=prompt,
             response_format="url",
-            provider='HuggingSpace'
+            provider='PollinationsAI'
         )
         
         image_url = response.data[0].url