cocktailpeanut commited on
Commit
63bc8dd
1 Parent(s): f1841f9
Files changed (2) hide show
  1. app.py +3 -3
  2. requirements.txt +2 -2
app.py CHANGED
@@ -4,7 +4,7 @@ from PIL import Image
4
  import numpy as np
5
  from aura_sr import AuraSR
6
  import torch
7
- import spaces
8
 
9
  # Force CPU usage
10
  torch.set_default_tensor_type(torch.FloatTensor)
@@ -19,7 +19,7 @@ aura_sr = AuraSR.from_pretrained("fal-ai/AuraSR")
19
  # Restore original torch.load
20
  torch.load = original_load
21
 
22
- @spaces.GPU
23
  def process_image(input_image):
24
  if input_image is None:
25
  return None
@@ -73,4 +73,4 @@ with gr.Blocks() as demo:
73
  cache_examples=True
74
  )
75
 
76
- demo.launch(debug=True)
 
4
  import numpy as np
5
  from aura_sr import AuraSR
6
  import torch
7
+ #import spaces
8
 
9
  # Force CPU usage
10
  torch.set_default_tensor_type(torch.FloatTensor)
 
19
  # Restore original torch.load
20
  torch.load = original_load
21
 
22
+ #@spaces.GPU
23
  def process_image(input_image):
24
  if input_image is None:
25
  return None
 
73
  cache_examples=True
74
  )
75
 
76
+ demo.launch(debug=True)
requirements.txt CHANGED
@@ -1,3 +1,3 @@
1
- spaces
2
  aura-sr
3
- gradio-imageslider
 
1
+ #spaces
2
  aura-sr
3
+ gradio-imageslider