Update app.py

app.py

@@ -60,7 +60,7 @@ def predict_image(image, model, device, transform, threshold=0.5):
 
 # Tải mô hình
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model = torch.load('faster-rcnn.pth')
+model = torch.load('faster-rcnn.pth', map_location=torch.device('cpu'))  # Thay đổi tại đây
 model.to(device).eval()
 
 # Định nghĩa hàm Gradio