Update app.py

app.py

@@ -7,9 +7,9 @@ import os
 
 os.system("wget https://github.com/liuxiaoyuyuyu/vanGogh-and-Other-Artist/blob/main/artist_classes.txt")
 
-model = torch.hub.load('pytorch/vision:v0.9.0', 'mobilenet_v2', pretrained=False)
-checkpoint = 'https://github.com/liuxiaoyuyuyu/vanGogh-and-Other-Artist/blob/main/model_weights_mobilenet_v2_valp1trainp2.pth'
-model.load_state_dict(torch.hub.load_state_dict_from_url(checkpoint, progress=False))
+model = torch.hub.load('pytorch/vision:v0.9.0', 'mobilenet_v2', pretrained=True)
+#checkpoint = 'https://github.com/liuxiaoyuyuyu/vanGogh-and-Other-Artist/blob/main/model_weights_mobilenet_v2_valp1trainp2.pth'
+#model.load_state_dict(torch.hub.load_state_dict_from_url(checkpoint, progress=False))
 model.eval()
 
 #torch.hub.download_url_to_file("https://github.com/pytorch/hub/raw/master/images/dog.jpg", "dog.jpg")