Update app.py

app.py

@@ -30,7 +30,7 @@ def get_seed(randomize_seed: bool, seed: int) -> int:
 #  The function for our Gradio app  #
 #####################################
 @spaces.GPU(duration=120)
-def generate(prompt, input_image, seed, progress=gr.Progress(track_tqdm=True)):
+def generate(prompt, input_image, seed):
     """
     Runs the Flux Control pipeline for editing the given `input_image`
     with the specified `prompt`. The pipeline is on GPU by default.
@@ -38,6 +38,9 @@ def generate(prompt, input_image, seed, progress=gr.Progress(track_tqdm=True)):
     seed = int(seed)  # Ensure seed is an integer
     generator = torch.Generator(device="cuda").manual_seed(seed)  # Maintain reproducibility
 
+    # Create a progress bar manually
+    progress = gr.Progress(track_tqdm=True)
+
     with progress.tqdm(total=1, desc="Generating Image") as pbar:
         output_image = pipeline(
             control_image=input_image,
@@ -115,4 +118,4 @@ def launch_app():
 
 if __name__ == "__main__":
     demo = launch_app()
-    demo.launch()
+    demo.launch()