Add files 3

app.py

@@ -33,7 +33,7 @@ def preprocess_image(image):
 # Prediction function
 def predict(image):
     # Load the input file
-    reloaded_img_tensor = torch.load(image).to(device)
+    reloaded_img_tensor = torch.load(image, map_location=device).to(device)  # Ensure tensor is loaded on the correct device
     
     # Make predictions
     output = model(reloaded_img_tensor)