Spaces:

antonovmaxim
/

KodIIm_14

Runtime error

antonovmaxim commited on May 7, 2023

Commit

f970b64

1 Parent(s): b5a5645

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ from torch import nn
 from torchvision import transforms
 from PIL import Image
 import os
-os.system(wget https://huggingface.co/antonovmaxim/aiornot-kodIIm-14/resolve/main/model.pth -O model.pth)
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 model = torch.hub.load("pytorch/vision", "resnet101", pretrained=False)
 model.fc = nn.Sequential(nn.Linear(2048, 500), nn.ReLU(), nn.Linear(500, 2), nn.Softmax(1))

 from torchvision import transforms
 from PIL import Image
 import os
+os.system("wget https://huggingface.co/antonovmaxim/aiornot-kodIIm-14/resolve/main/model.pth -O model.pth")
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 model = torch.hub.load("pytorch/vision", "resnet101", pretrained=False)
 model.fc = nn.Sequential(nn.Linear(2048, 500), nn.ReLU(), nn.Linear(500, 2), nn.Softmax(1))