Update app.py

app.py

@@ -16,6 +16,8 @@ from tempfile import NamedTemporaryFile
 # Create an FRCNN model instance with the same structure as the saved model
 model = torchvision.models.detection.fasterrcnn_resnet50_fpn(num_classes=91)
 
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+
 # Load the saved parameters into the model
 model.load_state_dict(torch.load("frcnn_model.pth"))