Update app.py

app.py

@@ -10,7 +10,7 @@ from sklearn.preprocessing import LabelEncoder
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
 # Load the pre-trained model
-model = torch.load('model.pt')
+model = torch.load('model.pt', map_location=torch.device('cpu'))
 model.eval()
 
 # Define the image transformation to normalize the image