sec

Dockerfile

@@ -29,6 +29,5 @@ RUN pip install --no-cache-dir -r requirements.txt
 
 COPY . .
 
-RUN where realesrgan-ncnn-vulkan
 
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -25,25 +25,24 @@ def home():
 @app.post("/enhance")
 async def enhance_image(file: UploadFile = File(...)):
     input_name = f"{uuid.uuid4()}_{file.filename}"
-    output_name = f"enhanced_{input_name}"
-
     input_path = os.path.join(OUTPUT_DIR, input_name)
-    output_path = os.path.join(OUTPUT_DIR, output_name)
 
     with open(input_path, "wb") as buffer:
         shutil.copyfileobj(file.file, buffer)
 
     cmd = [
-        REALESRGAN_PATH,
+        "realesrgan-ncnn-vulkan",
         "-i", input_path,
-        "-o", output_path,
-        "-s", "2"
+        "-o", OUTPUT_DIR,
+        "-s", "2",
+        "-m", "/opt/realesrgan/models"
     ]
 
-    subprocess.run(cmd, check=True)
+    result = subprocess.run(cmd, capture_output=True, text=True)
+
+    if result.returncode != 0:
+        raise HTTPException(status_code=500, detail=result.stderr)
+
+    enhanced = os.path.join(OUTPUT_DIR, f"enhanced_{input_name}")
 
-    return FileResponse(
-        output_path,
-        media_type="image/png",
-        filename=output_name
-    )
+    return FileResponse(enhanced, media_type="image/png")