Update app.py

app.py

@@ -6,16 +6,13 @@ import os
 
 torch.hub.download_url_to_file("https://github.com/pytorch/hub/raw/master/images/dog.jpg", "dog.jpg")
 
-model = torch.hub.load('pytorch/vision:v0.10.0', 'shufflenet_v2_x1_0', pretrained=True)
-model.load_state_dict(torch.load('shufflenetv2_x1-5666bf0f80.pth'))
-
  # To export
- model = torch.hub.load('pytorch/vision:v0.10.0', 'shufflenet_v2_x1_0', pretrained=True).eval()
- traced_graph = torch.jit.trace(model, torch.randn(1, 3, H, W))
- traced_graph.save('shufflenet.pth')
+model = torch.hub.load('pytorch/vision:v0.10.0', 'shufflenet_v2_x1_0', pretrained=True).eval()
+traced_graph = torch.jit.trace(model, torch.randn(1, 3, H, W))
+traced_graph.save('shufflenet.pth')
 
  # To load
- model = torch.jit.load('shufflenet.pth').eval().to(device)
+model = torch.jit.load('shufflenet.pth').eval().to(device)
 
 
 os.system("wget https://raw.githubusercontent.com/pytorch/hub/master/imagenet_classes.txt")