Update app.py

app.py

@@ -1,10 +1,10 @@
 import torch
 import torch.nn as nn
-from config import Net  # Make sure this matches your class name
+from config import MLP
 import gradio as gr
 
 # Load the model
-model = Net()
+model = MLP()
 model.load_state_dict(torch.load("pytorch_model.pth", map_location=torch.device("cpu")))
 model.eval()