vilarin commited on
Commit
efca2cc
·
verified ·
1 Parent(s): 6c44735

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -3,8 +3,11 @@ import spaces
3
  import gradio as gr
4
  from diffusers import FluxInpaintPipeline, FluxTransformer2DModel
5
  import random
 
6
  import numpy as np
 
7
 
 
8
  MAX_SEED = np.iinfo(np.int32).max
9
  model = "black-forest-labs/FLUX.1-dev"
10
  hf_hub_download(repo_id="black-forest-labs/FLUX.1-Fill-dev", filename="ae.safetensors", local_dir=".")
 
3
  import gradio as gr
4
  from diffusers import FluxInpaintPipeline, FluxTransformer2DModel
5
  import random
6
+ import os
7
  import numpy as np
8
+ from huggingface_hub import hf_hub_download
9
 
10
+ os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
11
  MAX_SEED = np.iinfo(np.int32).max
12
  model = "black-forest-labs/FLUX.1-dev"
13
  hf_hub_download(repo_id="black-forest-labs/FLUX.1-Fill-dev", filename="ae.safetensors", local_dir=".")