Update app.py

app.py

@@ -206,7 +206,7 @@ from PIL import Image
 import torchvision.transforms as T
 
 # Load your model
-model = torch.load("model.pt")
+model = torch.load("Model_IV.pt")
 model.eval()
 
 # Define preprocessing