Update main.py

main.py

@@ -198,11 +198,17 @@ class UpscaleRequest(BaseModel):
 def process_image(image_url):
     response = requests.get(image_url)
     img = Image.open(BytesIO(response.content))
-    upscaled_image = aura_sr.upscale_4x_overlapped(img)
+    if aura_sr is None:
+        raise RuntimeError("aura_sr is None. Check initialization.")
+    try:
+        upscaled_image = aura_sr.upscale_4x_overlapped(img)
+    except Exception as e:
+        raise RuntimeError(f"Error during upscaling: {str(e)}")
     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):
     # Add the task directly to the background