Update app.py

app.py

@@ -8,7 +8,7 @@ from diffusers.utils import load_image
 
 from huggingface_hub import login
 hf_token = os.environ.get("HF_TOKEN")
-login(hf_token)
+login(token=hf_token)
 
 def load_pipeline(four_bit=False):
     orig_pipeline = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
@@ -42,7 +42,7 @@ def get_canny(control_image):
     )
     return control_image
 
-def main(ref_filepath, prompt, use_nf4):
+def main(ref_filepath, prompt, use_nf4, progress=gr.Progress(track_tqdm=True)):
     pipe = load_pipeline(use_nf4)
     control_image = load_image(ref_filepath)
     control_image = get_canny(control_image)