Update app.py

app.py

@@ -1,6 +1,5 @@
 import torch
-model = torch.hub.load('pytorch/vision:v0.9.0', 'fcn_resnet101', pretrained=True)
-model.eval()
+
 # Download an example image from the pytorch website
 import urllib
 url, filename = ("https://github.com/pytorch/hub/raw/master/images/dog.jpg", "dog.jpg")
@@ -13,6 +12,10 @@ import gradio as gr
 import matplotlib.pyplot as plt
 
 
+
+model = torch.hub.load('pytorch/vision:v0.9.0', 'fcn_resnet101', pretrained=True)
+
+
 def inference(input_image):
     preprocess = transforms.Compose([
         transforms.ToTensor(),