Spaces:
Runtime error
Runtime error
sunshangquan
commited on
Commit
•
14e98b6
1
Parent(s):
0e065e1
commit from ssq
Browse files- app.py +1 -1
- requirements.txt +1 -0
app.py
CHANGED
|
@@ -7,7 +7,7 @@ from skimage import img_as_ubyte
|
|
| 7 |
from Allweather.util import load_img, save_img
|
| 8 |
from basicsr.models.archs.histoformer_arch import Histoformer
|
| 9 |
|
| 10 |
-
model_restoration = Histoformer.from_pretrained("sunsean/Histoformer-real")
|
| 11 |
|
| 12 |
model_restoration.eval()
|
| 13 |
|
|
|
|
| 7 |
from Allweather.util import load_img, save_img
|
| 8 |
from basicsr.models.archs.histoformer_arch import Histoformer
|
| 9 |
|
| 10 |
+
model_restoration = Histoformer.from_pretrained("sunsean/Histoformer-real").to("cuda")
|
| 11 |
|
| 12 |
model_restoration.eval()
|
| 13 |
|
requirements.txt
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
|
|
| 1 |
torch
|
| 2 |
torchvision
|
| 3 |
scikit-image
|
|
|
|
| 1 |
+
--extra-index-url https://download.pytorch.org/whl/cu113
|
| 2 |
torch
|
| 3 |
torchvision
|
| 4 |
scikit-image
|