Update app.py

app.py

@@ -3,7 +3,7 @@ import torch
 import torch.nn as nn
 import os
 
-# the model architecture
+# architecture
 class AddModel(nn.Module):
     def __init__(self):
         super(AddModel, self).__init__()
@@ -19,24 +19,23 @@ class AddModel(nn.Module):
         x = self.fc3(x)
         return x
 
-# load the model from a specified path
 def load_model(model_path):
     model = AddModel()
     model.load_state_dict(torch.load(model_path, map_location=torch.device('cpu')))
-    model.eval()  # Set the model to evaluation mode
+    model.eval()  # evaluation mode
     return model
 
-#predictions
 def predict_sum(model, x1, x2):
     with torch.no_grad():
         input_tensor = torch.tensor([[x1, x2]], dtype=torch.float32)
         prediction = model(input_tensor)
         return prediction.item()
 
+# Streamlit app
 def main():
     st.title("Sum Predictor using Neural Network")
-    
-    model_path = "./models/MA1T.pth"
+
+    model_path = "MA1T.pth"  # Update with your model path if necessary
     if os.path.exists(model_path):
         model = load_model(model_path)
         st.success("Model loaded successfully.")