thak123 commited on
Commit
fc66215
1 Parent(s): 2e53777

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -9,6 +9,8 @@ import gradio as gr
9
  import torch
10
  import torch.nn as nn
11
  from transformers import CLIPModel, AutoModel
 
 
12
 
13
  from huggingface_hub import hf_hub_download
14
  from safetensors.torch import load_model
 
9
  import torch
10
  import torch.nn as nn
11
  from transformers import CLIPModel, AutoModel
12
+ from torchvision.transforms import CenterCrop, ConvertImageDtype, Normalize, Resize
13
+ from torchvision.transforms.functional import InterpolationMode
14
 
15
  from huggingface_hub import hf_hub_download
16
  from safetensors.torch import load_model