updated ckpt and app.py

app.py

@@ -193,7 +193,7 @@ if __name__ == "__main__":
     }
 
     DEVICE = torch.device("cuda:0") if torch.cuda.is_available() else torch.device("cpu")
-    CKPT_PATH = os.path.join(os.getcwd(), r"ckpt_024-vloss_0.1816_vf1_0.7855.ckpt")
+    CKPT_PATH = os.path.join(os.getcwd(), r"ckpt_022-vloss_0.1756_vf1_0.7919.ckpt")
     model = ProteinModel.load_from_checkpoint(CKPT_PATH)
     model.to(DEVICE)
     model.eval()
@@ -208,7 +208,7 @@ if __name__ == "__main__":
     )
 
     images_dir = glob(os.path.join(os.getcwd(), "samples") + os.sep + "*.png")
-    examples = [[i, TrainingConfig.METRIC_THRESH] for i in np.random.choice(images_dir, size=8, replace=False)]
+    examples = [[i, TrainingConfig.METRIC_THRESH] for i in np.random.choice(images_dir, size=10, replace=False)]
     # print(examples)
 
     iface = gr.Interface(

ckpt_024-vloss_0.1816_vf1_0.7855.ckpt → ckpt_022-vloss_0.1756_vf1_0.7919.ckpt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:eeba4764adb310bf3a35ba2479326fdbf38acaed3242a9f020ff2d7eba47b2ca
+oid sha256:8f0bb009e4d3c07380ed58b5078df7ec08f8adccd742b44aff99b4b35531300e
 size 243578302