added auraflow autopiope

check_app.py

@@ -91,18 +91,21 @@ def generate_image_with_progress(pipe, prompt, num_steps, guidance_scale=None, s
 @spaces.GPU(duration=170)
 def create_pipeline_logic(prompt_text, model_name):
     print(f"starting {model_name}")
-    progress = gr.Progress()
+    progress = gr.Progress(track_tqdm=True)
     num_steps = 30
     guidance_scale = 7.5  # Example guidance scale, can be adjusted per model
     seed = 42
     config = MODEL_CONFIGS[model_name]
     pipe_class = config["pipeline_class"]
     pipe = None
-    if model_name == "Kandinsky":
-        print("Kandinsky Special")
+    if model_name == "Kandinsky" or model_name == "AuraFlow":
+        print("Kandinsky or AuraFlow Special")
         pipe = AutoPipelineForText2Image.from_pretrained(
-            "kandinsky-community/kandinsky-3", variant="fp16", torch_dtype=torch.float16
-        )
+            config["repo_id"],
+            variant="fp16",
+            #cache_dir=config["cache_dir"],
+            torch_dtype=torch.bfloat16
+        ).to("cuda")
     else:
         pipe = pipe_class.from_pretrained(
             config["repo_id"],