Update app.py

app.py

@@ -21,7 +21,7 @@ class RegressionModel2(nn.Module):
         return out
 
 # Load the saved model state dictionary
-model = RegressionModel2(input_dim2, hidden_dim2, output_dim2)
+model = RegressionModel2(3, 32, 1)
 model.load_state_dict(torch.load('model.pt'))
 model.eval()  # Set the model to evaluation mode