Update app.py

app.py

@@ -150,8 +150,8 @@ def predict_raw(raw_input):
     _, dataloader = generate_dataset_dataloader(ids, seqs)
 
     model = CNN_linear()
-    print(model.state_dict().keys())
-    print(torch.load(modelfile, map_location=torch.device('cpu')).keys())
+    st.write(model.state_dict().keys())
+    st.write(torch.load(modelfile, map_location=torch.device('cpu')).keys())
     model.esm2.load_state_dict({k.replace('module.', ''):v for k,v in torch.load(modelfile, map_location=torch.device('cpu')).items()}, strict = False)
     # model.load_state_dict(torch.load(modelfile, map_location=torch.device('cpu')), strict = False)