Update app.py

app.py

@@ -3,6 +3,7 @@ from diffusers import DiffusionPipeline
 import spaces
 import torch
 from concurrent.futures import ProcessPoolExecutor
+from huggingface_hub import hf_hub_download
 
 dev_model = "black-forest-labs/FLUX.1-dev"
 schnell_model = "black-forest-labs/FLUX.1-schnell"
@@ -20,6 +21,7 @@ pipe_schnell = DiffusionPipeline.from_pretrained(
 )
 @spaces.GPU
 def run_dev_hyper(prompt):
+    print("dev_hyper")
     pipe_dev.to("cuda")
     repo_name = "ByteDance/Hyper-SD"
     ckpt_name = "Hyper-FLUX.1-dev-8steps-lora.safetensors"
@@ -30,6 +32,7 @@ def run_dev_hyper(prompt):
 
 @spaces.GPU
 def run_dev_turbo(prompt):
+    print("dev_turbo")
     pipe_dev.to("cuda")
     repo_name = "alimama-creative/FLUX.1-Turbo-Alpha"
     ckpt_name = "diffusion_pytorch_model.safetensors"
@@ -40,6 +43,7 @@ def run_dev_turbo(prompt):
 
 @spaces.GPU
 def run_schnell(prompt):
+    print("schnell")
     pipe_schnell.to("cuda")
     image = pipe_schnell(prompt).images[0]
     return image