Hugging Face's logo

Spaces:
Jfink09
/
Meyestro_TCA_Prediction
Running

App Files Community
Jfink09 commited on
Commit
dcaeb83
·
verified ·
1 Parent(s): 56254a7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ class RegressionModel2(nn.Module):
20
 
21
  # Load the saved model state dictionaries
22
  model_j0 = RegressionModel2(3, 32, 1)
23
- model_j0.load_state_dict(torch.load('j0_model.pt'))
24
  model_j0.eval()
25
 
26
  model_j45 = RegressionModel2(3, 32, 1)
 
20
 
21
  # Load the saved model state dictionaries
22
  model_j0 = RegressionModel2(3, 32, 1)
23
+ model_j0.load_state_dict(torch.load('j0_model-2.pt'))
24
  model_j0.eval()
25
 
26
  model_j45 = RegressionModel2(3, 32, 1)