Update app.py

app.py

@@ -6,8 +6,7 @@ from torchvision import transforms
 from PIL import Image
 
 # Load your PyTorch model
-model = torch.load('your_model.pth')  # Replace with your model path
-model.eval()
+model = torch.hub.load('pytorch/vision:v0.6.0', 'resnet18', pretrained=True).eval()
 
 # Define a function to preprocess the DICOM
 def preprocess_dicom(dicom_path):