Update app.py

app.py

@@ -6,6 +6,7 @@ model = torchvision.models.resnet50(pretrained=False)
 model.fc = nn.Linear(model.fc.in_features, 2)
 model.load_state_dict(torch.load("model.pth", map_location=torch.device('cpu')))
 model.eval()
+device = torch.device("cpu")
 
 import gradio as gr
 from PIL import Image