fix diffuser

app.py

@@ -1,7 +1,7 @@
-from fastapi import FastAPI, File, UploadFile, HTTPException
-from fastapi.responses import StreamingResponse
-from diffusers import MarigoldDepthPipeline
+import diffusers
 import torch
+from fastapi import FastAPI, UploadFile, HTTPException, File
+from fastapi.responses import StreamingResponse
 from PIL import Image
 import io
 
@@ -33,18 +33,18 @@ async def predict_depth(file: UploadFile = File(...)):
 
         # Visualiza la profundidad
         vis = pipe.image_processor.visualize_depth(depth.prediction)
-         # Guarda la imagen generada en un buffer
+
+        # Guarda la imagen generada en un buffer
         img_buffer = io.BytesIO()
         vis[0].save(img_buffer, format="PNG")
-        img_buffer.seek(0)  # Regresa el cursor al inicio
+        img_buffer.seek(0)
 
-        # Retorna la imagen como respuesta
+        # Devuelve la imagen como respuesta
         return StreamingResponse(img_buffer, media_type="image/png")
-
     except Exception as e:
         print(f"[ERROR] {str(e)}")
+        raise HTTPException(status_code=500, detail="Error procesando la imagen.")
 
-    
 @app.get("/")
 async def root():
     return {"message": "API de generación de mapas de profundidad con Marigold"}