Spaces:
Running
on
Zero
Running
on
Zero
Anton Bushuiev
commited on
Commit
·
76d7f2a
1
Parent(s):
53691e3
Minor update
Browse files- app.py +3 -3
- requirements.txt +0 -1
app.py
CHANGED
|
@@ -486,9 +486,9 @@ with app:
|
|
| 486 |
# Download weights from Zenodo
|
| 487 |
download_weights()
|
| 488 |
|
| 489 |
-
#
|
| 490 |
-
|
| 491 |
-
device = torch.device('cuda')
|
| 492 |
|
| 493 |
# Load models
|
| 494 |
models = [
|
|
|
|
| 486 |
# Download weights from Zenodo
|
| 487 |
download_weights()
|
| 488 |
|
| 489 |
+
# Set device
|
| 490 |
+
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
| 491 |
+
# device = torch.device('cuda')
|
| 492 |
|
| 493 |
# Load models
|
| 494 |
models = [
|
requirements.txt
CHANGED
|
@@ -1,3 +1,2 @@
|
|
| 1 |
ppiformer @ git+https://github.com/anton-bushuiev/ppiformer.git@main
|
| 2 |
gradio
|
| 3 |
-
# gradio==4.17.0
|
|
|
|
| 1 |
ppiformer @ git+https://github.com/anton-bushuiev/ppiformer.git@main
|
| 2 |
gradio
|
|
|