Update app.py

app.py

@@ -2,8 +2,9 @@ import gradio as gr
 import torchvision
 import torchvision.transforms as transforms
 import torch
+from huggingface_hub import load_model_from_hub
 
-model = torch.jit.load('https://huggingface.co/spaces/1-13-am/neural-style-transfer/blob/main/neural_style_transfer.pt')
+model = load_model_from_hub("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]
@@ -17,13 +18,8 @@ def denormalize():
   MEAN = [-mean/std for mean, std in zip(MEAN, STD)]
   STD = [1/std for std in STD]
   return transforms.Normalize(mean=MEAN, std=STD)
-def transformer(imsize = None, cropsize = None):
+def transformer():
   transformer = []
-  if imsize:
-        transformer.append(transforms.Resize(imsize))
-  if cropsize:
-        transformer.append(transforms.RandomCrop(cropsize))
-
   transformer.append(transforms.ToTensor())
   transformer.append(normalize())
   return transforms.Compose(transformer)