Update app.py

app.py

@@ -18,6 +18,10 @@ class RegressionModel2(nn.Module):
         out = self.fc2(out)
         return out
 
+input_dim2 = X2_train.shape[1] # change to [1] for pentacam dataset X_train.shape[1]
+hidden_dim2 = 32 # Was 16
+output_dim2 = 1
+
 # Load your saved model state dictionary (assuming 'model.pt' is uploaded)
 model2 = RegressionModel2(input_dim2, hidden_dim2, output_dim2)
 model2.load_state_dict(torch.load('model.pt'))