Update app.py

app.py

@@ -15,7 +15,7 @@ os.system("wget https://github.com/liuxiaoyuyuyu/vanGogh-and-Other-Artist/blob/m
 model = models.mobilenet_v2(pretrained=True)
 num_ftrs = model.classifier[1].in_features
 model.classifier[1] = nn.Linear(num_ftrs, 6)
-model = model.to(device)
+#model = model.to(device)
 #model.load_state_dict(torch.load('model_weights_mobilenet_v2_valp1trainp2.pth'))
 
 #torch.hub.download_url_to_file("https://github.com/pytorch/hub/raw/master/images/dog.jpg", "dog.jpg")