Update app.py

app.py

@@ -3,16 +3,18 @@ import os
 os.environ["WANDB_DISABLED"] = "true"
 
 import numpy as np
-
 from PIL import Image
+
 import gradio as gr
 import torch
 import torch.nn as nn
-from transformers import CLIPModel, AutoModel
+
+import torchvision
 from torchvision.transforms import CenterCrop, ConvertImageDtype, Normalize, Resize
 from torchvision.transforms.functional import InterpolationMode
 from torchvision import transforms
 
+from transformers import CLIPModel, AutoModel
 from huggingface_hub import hf_hub_download
 from safetensors.torch import load_model