Update app.py

app.py

@@ -21,6 +21,9 @@ from plot_pdb import plot_struc
 HF_TOKEN = os.environ.get("HF_token")
 print("HF_TOKEN:",HF_TOKEN)
 
+device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+print("Running device is:", device)
+
 MODEL_OPTIONS = [
     "facebook/esm2_t6_8M_UR50D",
     "facebook/esm2_t12_35M_UR50D",