update

app.py

@@ -5,6 +5,8 @@ from PIL import Image
 import numpy as np
 from aura_sr import AuraSR
 import torch
+import devicetorch
+DEVICE = devicetorch.get(torch)
 
 # Force CPU usage
 torch.set_default_tensor_type(torch.FloatTensor)
@@ -14,7 +16,7 @@ original_load = torch.load
 torch.load = lambda *args, **kwargs: original_load(*args, **kwargs, map_location=torch.device('cpu'))
 
 # Initialize the AuraSR model
-aura_sr = AuraSR.from_pretrained("fal/AuraSR-v2")
+aura_sr = AuraSR.from_pretrained("fal/AuraSR-v2", device=DEVICE)
 
 # Restore original torch.load
 torch.load = original_load
@@ -72,7 +74,7 @@ with gr.Blocks() as demo:
         inputs=input_image,
         outputs=output_slider,
         fn=process_image,
-        cache_examples=True
+#        cache_examples=True
     )
 
-demo.launch(debug=True)
+demo.launch(debug=True)

requirements.txt

@@ -1,3 +1,3 @@
 spaces
-aura-sr
-gradio-imageslider
+aura-sr@git+https://github.com/cocktailpeanut/aura-sr.git
+gradio-imageslider