Update app.py

app.py

@@ -9,7 +9,7 @@ from torchvision import transforms
 model = torch.hub.load(
     "RF5/danbooru-pretrained",
     "resnet50",
-    map_location=torch.device("cpu"),
+    map_location="cpu",
 )
 model.eval()
 
@@ -53,4 +53,4 @@ threshold = inputs.Slider(
 labels = outputs.Label(type="confidence")
 
 interface = gradio.Interface(main, inputs=[image, threshold], outputs=[labels])
-interface.launch()
+interface.launch()