Update app.py

app.py

@@ -24,11 +24,11 @@ class RegressionModel2(nn.Module):
 @st.cache_resource
 def load_models():
     model_j0 = RegressionModel2(3, 32, 1)
-    model_j0.load_state_dict(torch.load('j0_model.pt'))
+    model_j0.load_state_dict(torch.load('j0_model-2.pt'))
     model_j0.eval()
 
     model_j45 = RegressionModel2(3, 32, 1)
-    model_j45.load_state_dict(torch.load('j45_model.pt'))
+    model_j45.load_state_dict(torch.load('j45_model-2.pt'))
     model_j45.eval()
     
     return model_j0, model_j45