Spaces:
Running
on
T4
Running
on
T4
simonduerr
commited on
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
|