Update app.py

app.py

@@ -1,5 +1,6 @@
 import torch
 import gradio as gr
+import torchvision.transforms as transforms
 
 device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
 
@@ -12,7 +13,7 @@ def inference(img):
 
   img_transforms = transforms.Compose(
                 [transforms.Resize(256), transforms.CenterCrop(224), transforms.ToTensor()]
-            )
+  )
 
   img = img_transforms(img)
   with torch.no_grad():