Update app.py

app.py

@@ -82,7 +82,7 @@ class ResnUnet(nn.Module):
 model=ResnUnet().to('cpu')
 
 # Load the state_dict
-state_dict = torch.load('/content/real_model1_onnx_compat.pt',map_location='cpu')
+state_dict = torch.load('real_model2_onnx_compat.pt',map_location='cpu')
 
 # Create a new state_dict without the 'module.' prefix
 new_state_dict = {}