Update app.py

app.py

@@ -1,6 +1,9 @@
 import gradio as gr
 import torchvision
 import torchvision.transforms as transforms
+import torch
+
+model = torch.jit.load('https://huggingface.co/spaces/1-13-am/neural-style-transfer/blob/main/neural_style_transfer.pt')
 
 def normalize():
   MEAN = [0.485, 0.456, 0.406]