simonduerr commited on
Commit
b4ecf2a
1 Parent(s): d13186d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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