Update app.py

app.py

@@ -1,11 +1,11 @@
+import os
+os.system('pip install --upgrade torch torchvision')
+os.system("wget https://huggingface.co/antonovmaxim/aiornot-kodIIm-14/resolve/main/model.pth -O model.pth")
 import gradio as gr
 import torch
 from torch import nn
 from torchvision import transforms
 from PIL import Image
-import os
-os.system('pip install --upgrade torch torchvision')
-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))