Update main.py

main.py

@@ -195,22 +195,23 @@ class UpscaleRequest(BaseModel):
     image_url: str
 
 
+def process_image(image_url):
+    response = requests.get(image_url)
+    img = Image.open(BytesIO(response.content))
+    upscaled_image = aura_sr.upscale_4x_overlapped(img)
+    filename = f"upscaled_{uuid.uuid4()}.png"
+    filepath = save_image_locally(upscaled_image, filename)
+    return filepath
+
 @app.post("/upscale-image", response_model=dict)
 async def upscale_image(request: UpscaleRequest, background_tasks: BackgroundTasks):
-    def process_image():
-        response = requests.get(request.image_url)
-        img = Image.open(BytesIO(response.content))
-        upscaled_image = aura_sr.upscale_4x_overlapped(img)
-        filename = f"upscaled_{uuid.uuid4()}.png"
-        filepath = save_image_locally(upscaled_image, filename)
-        return filepath
-
-    task = executor.submit(process_image)
-    background_tasks.add_task(task)
+    # Add the task directly to the background
+    background_tasks.add_task(process_image, request.image_url)
     return {"status": "Processing"}
 
 
 
+
 @app.get("/")
 async def root():
     return {"message": "API is up and running!"}