Update app.py

app.py

@@ -4,8 +4,8 @@ import gradio as gr
 	
 	# Cargamos el learner
 model = models.torchvision.faster_rcnn.model(backbone=models.torchvision.faster_rcnn.backbones.resnet18_fpn,
-                                             num_classes=2)
-state_dict = torch.load('fasterRCNNkangaroo.pth')
+                                             num_classes=1)
+state_dict = torch.load('fasterRCNNkangaroo.pth', map_location = torch.device('cpu'))
 model.load_state_dict(state_dict)