Update app.py

app.py

@@ -10,6 +10,7 @@ from translatepy import Translator
 #from huggingface_hub import hf_hub_download
 import requests
 import re
+import asyncio
 
 os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
 translator = Translator()
@@ -44,7 +45,7 @@ def enable_lora(lora_in, lora_add):
         return lora_in
 
 @spaces.GPU()
-def generate_image(
+async def generate_image(
     prompt:str,
     model:str,
     width:int=768,
@@ -83,7 +84,7 @@ def generate_image(
 
     return image1, image2, seed
 
-def gen(
+async def gen(
     prompt:str,
     lora_in:str="", 
     lora_add:str="",
@@ -95,7 +96,8 @@ def gen(
     progress=gr.Progress(track_tqdm=True)
 ):
     model = enable_lora(lora_in, lora_add)
-    return generate_image(prompt,model,width,height,scales,steps,seed)
+    image1, image2, seed = await generate_image(prompt,model,width,height,scales,steps,seed)
+    return image1, image2, seed