update sync calling generation

app.py

@@ -14,7 +14,6 @@ import gc
 from io import BytesIO
 import base64
 import functools
-import asyncio
 
 app = FastAPI()
 
@@ -120,8 +119,8 @@ class ModelContext:
 
 model_manager = ModelManager()
 
-# @spaces.GPU
-async def generate_image(
+@spaces.GPU
+def generate_image(
     model_name: str,
     prompt: str,
     height: int = 512,
@@ -171,7 +170,6 @@ async def generate_image(
         model_manager.unload_current_model()
         raise HTTPException(status_code=500, detail=str(e))
 
-
 @app.post("/generate")
 async def generate_image_endpoint(
     model_name: str,
@@ -187,7 +185,7 @@ async def generate_image_endpoint(
         content = await reference_image.read()
         ref_img = Image.open(BytesIO(content))
     
-    return await generate_image(
+    return generate_image(
         model_name=model_name,
         prompt=prompt,
         height=height,
@@ -206,7 +204,6 @@ async def unload_model():
     model_manager.unload_current_model()
     return {"status": "success", "message": "Model unloaded"}
 
-
 def create_gradio_interface() -> gr.Blocks:
     with gr.Blocks() as interface:
         gr.Markdown("# Text-to-Image Generation Interface")
@@ -300,8 +297,8 @@ def create_gradio_interface() -> gr.Blocks:
                 )
             ]
         
-        async def generate(model_name: str, prompt_text: str, h: int, w: int, steps: int, guide_scale: float, ref_img: Optional[Image.Image]) -> Image.Image:
-            response = await generate_image(
+        def generate(model_name: str, prompt_text: str, h: int, w: int, steps: int, guide_scale: float, ref_img: Optional[Image.Image]) -> Image.Image:
+            response = generate_image(
                 model_name=model_name,
                 prompt=prompt_text,
                 height=h,