Update app.py

app.py

@@ -80,6 +80,8 @@ def load_model(model_name: str):
             model_seq,
             new_head
         )
+        weights = torch.load('/CNN_1stGEAC_m2_best.pth')
+        model.load_state_dict(weights)
         return model, None
 
     else: