Update app.py

app.py

@@ -17,7 +17,7 @@ class ModelActor:
         """
         self.model = AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo")
 
-    def generate_image(self, prompt, prompt_name):
+    async def generate_image(self, prompt, prompt_name):
         """
         Generates an image based on the provided prompt.
         Parameters:
@@ -29,7 +29,7 @@ class ModelActor:
         start_time = time.time()
         process_id = os.getpid()
         try:
-            output = self.model(prompt=prompt, num_inference_steps=1, guidance_scale=0.0)
+            output = await self.model(prompt=prompt, num_inference_steps=1, guidance_scale=0.0)
             if isinstance(output.images, list) and len(output.images) > 0:
                 image = output.images[0]
                 buffered = BytesIO()
@@ -93,4 +93,4 @@ gradio_interface = gr.Interface(
 )
 
 if __name__ == "__main__":
-    gradio_interface.launch()
+    gradio_interface.launch()