Spaces:
Runtime error
Runtime error
cocktailpeanut
commited on
Commit
•
63bc8dd
1
Parent(s):
f1841f9
update
Browse files- app.py +3 -3
- 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 |
-
|
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
|