gabar92 commited on
Commit
698149b
·
1 Parent(s): f74866d
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -8,6 +8,7 @@ from functools import partial
8
 
9
  import gradio as gr
10
  from PIL import Image
 
11
 
12
  from die_model import UNetDIEModel
13
  from utils import resize_image, make_image_square, cast_pil_image_to_torch_tensor_with_4_channel_dim, \
 
8
 
9
  import gradio as gr
10
  from PIL import Image
11
+ from huggingface_hub import hf_hub_download
12
 
13
  from die_model import UNetDIEModel
14
  from utils import resize_image, make_image_square, cast_pil_image_to_torch_tensor_with_4_channel_dim, \