Update app.py

app.py

@@ -5,7 +5,7 @@ import socket
 import requests
 import sys
 import logging
-from fastapi import FastAPI, File, UploadFile, Form
+from fastapi import FastAPI, File, UploadFile, Form, BackgroundTasks
 from fastapi.responses import FileResponse, StreamingResponse
 from fastapi.middleware.cors import CORSMiddleware
 from PIL import Image
@@ -272,8 +272,9 @@ async def generate_image(
     seed: int = Form(...),
     sampler_type: str = Form(...),
     image: UploadFile = File(...)
+    background_tasks: BackgroundTasks
 ):
-    def generate_image_task():
+    async def generate_image_task():
         try:
             # Save the uploaded image to a temporary file
             temp_image_path = f"/tmp/{int(time.time())}_{image.filename}"
@@ -301,7 +302,7 @@ async def generate_image(
             return "Failed to generate image"
 
     try:
-        queue_manager.add_task(generate_image_task)
+        background_tasks.add_task(generate_image_task)
         position_in_queue = queue_manager.queue.qsize()
         # Total queue size is still async
         total_queue_size = await queue_manager.get_total_queue_size()  # Implement this function