Update app.py

app.py

@@ -309,24 +309,27 @@ if huggingface_token:
 else:
     print("Hugging Face token not found in environment variables.")
 print(huggingface_token)
-model_path = snapshot_download(
+with tqdm(total=100, desc="Downloading model", bar_format="{l_bar}{bar}| {n_fmt}/{total_fmt}") as pbar:
+ model_path = snapshot_download(
     repo_id="black-forest-labs/FLUX.1-dev", 
     repo_type="model", 
     ignore_patterns=["*.md", "*..gitattributes"],
     local_dir="FLUX.1-dev",
-    token=huggingface_token
+    token=huggingface_token)
 
 # Load pipeline
 print('controlnet enters')
-controlnet = FluxControlNetModel.from_pretrained(
+with tqdm(total=100, desc="Downloading model", bar_format="{l_bar}{bar}| {n_fmt}/{total_fmt}") as pbar:
+ controlnet = FluxControlNetModel.from_pretrained(
     "jasperai/Flux.1-dev-Controlnet-Upscaler", torch_dtype=torch.bfloat16
 ).to(device)
 print('controlnet exits')
 print('pipe enters')
-pipe = FluxControlNetPipeline.from_pretrained(
+with tqdm(total=100, desc="Downloading model", bar_format="{l_bar}{bar}| {n_fmt}/{total_fmt}") as pbar:
+ pipe = FluxControlNetPipeline.from_pretrained(
     model_path, controlnet=controlnet, torch_dtype=torch.bfloat16
-)
-pipe.to(device)
+).to(device)
+# pipe.to(device)
 print('pipe exits')
 
 MAX_SEED = 1000000