Update

app.py

@@ -119,6 +119,7 @@ def build_ui():
     print(f"Initializing model on {device}...")
     _get_model(device)
 
+    @spaces.GPU
     def run_inference(image: np.ndarray | None) -> np.ndarray | None:
         """Run reflection removal using the cached model. Returns RGB numpy [H,W,3] in 0–255 or None."""
         if image is None:

requirements.txt

@@ -1,5 +1,4 @@
 
 gradio
 huggingface-hub
-unreflectanything @ git+https://github.com/alberto-rota/UnReflectAnything.git
-rich
+unreflectanything @ git+https://github.com/alberto-rota/UnReflectAnything.git