Update app.py

app.py

@@ -1,28 +1,25 @@
 import os
 import asyncio
 
-import prodiapy
+import prodiapy.aio  # Import the async module
 import gradio as gr
 
 # Load API key from Hugging Face Secrets
-api_key = os.getenv("PRODIA_AI_API")  
-prodia = prodiapy.Prodia(api_key=api_key)
+api_key = os.getenv("PRODIA_AI_API") 
+prodia = prodiapy.aio.Prodia(api_key=api_key)  # Initialize async client
 
 async def generate_image(prompt):
-    job = await prodia.sd.async_generate(prompt=prompt)
-    result = await prodia.async_wait(job)
+    job = await prodia.sd.generate(prompt=prompt)
+    result = await prodia.wait(job)
     return result.image_url
 
-async def process_image(prompt):
-    return await asyncio.to_thread(generate_image, prompt) # Run in a thread
-
 demo = gr.Interface(
-    fn=process_image,
+    fn=generate_image,  
     inputs="text",
     outputs="image",
     title="Async Prodia Image Generation",
     description="Generate images asynchronously using Prodia API.",
 )
 
-demo.queue() # Enable queuing to handle multiple requests
-demo.launch()
+demo.queue() 
+demo.launch()