Spaces:
Running
on
T4
Running
on
T4
Commit
·
b4ecf2a
1
Parent(s):
d13186d
Update app.py
Browse files
app.py
CHANGED
|
@@ -74,6 +74,8 @@ def clear_mem():
|
|
| 74 |
buf.delete()
|
| 75 |
|
| 76 |
|
|
|
|
|
|
|
| 77 |
def setup_af(seq, model_name="model_5_ptm"):
|
| 78 |
clear_mem()
|
| 79 |
# setup model
|
|
|
|
| 74 |
buf.delete()
|
| 75 |
|
| 76 |
|
| 77 |
+
print("Is cuda available",torch.cuda.is_available)
|
| 78 |
+
|
| 79 |
def setup_af(seq, model_name="model_5_ptm"):
|
| 80 |
clear_mem()
|
| 81 |
# setup model
|