Update app.py

app.py

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