Hugging Face's logo

Spaces:
Jfink09
/
Meyestro_TCA_Prediction
Running

App Files Community
Jfink09 commited on
Commit
a5bcb7f
·
verified ·
1 Parent(s): 264b03d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ model_j0.load_state_dict(torch.load('j0_model-2.pt'))
24
  model_j0.eval()
25
 
26
  model_j45 = RegressionModel2(3, 32, 1)
27
- model_j45.load_state_dict(torch.load('j45_model.pt'))
28
  model_j45.eval()
29
 
30
  def calculate_initial_j0_j45(magnitude, axis_deg):
 
24
  model_j0.eval()
25
 
26
  model_j45 = RegressionModel2(3, 32, 1)
27
+ model_j45.load_state_dict(torch.load('j45_model-2.pt'))
28
  model_j45.eval()
29
 
30
  def calculate_initial_j0_j45(magnitude, axis_deg):