change func

check_app.py

@@ -14,7 +14,7 @@ from diffusers import (
 import gradio as gr
 from diffusers.pipelines.pipeline_utils import DiffusionPipeline
 
-class ProgressAuraFlowPipeline(DiffusionPipeline):
+class ProgressPipeline(DiffusionPipeline):
     def __init__(self, original_pipeline):
         super().__init__()
         self.original_pipeline = original_pipeline
@@ -164,6 +164,8 @@ def generate_image_with_progress(model_name,pipe, prompt, num_steps, guidance_sc
             prompt,
             num_inference_steps=num_steps,
             generator=generator,
+            callback=callback,
+            callback_steps=1,
         ).images[0]
 
     return image
@@ -186,11 +188,11 @@ def create_pipeline_logic(prompt_text, model_name):
             #cache_dir=config["cache_dir"],
             torch_dtype=torch.bfloat16
         ).to("cuda")
-        pipe = ProgressAuraFlowPipeline(b_pipe)
+        pipe = ProgressPipeline(b_pipe)
     else:
         pipe = AutoPipelineForText2Image.from_pretrained(
             config["repo_id"],
-            variant="fp16",
+            #variant="fp16",
             #cache_dir=config["cache_dir"],
             torch_dtype=torch.bfloat16
         ).to("cuda")