added flux

check_app.py

@@ -15,6 +15,11 @@ import gradio as gr
 cache_dir = '/workspace/hf_cache'
 
 MODEL_CONFIGS = {
+        "FLUX": {
+        "repo_id": "black-forest-labs/FLUX.1-dev",
+        "pipeline_class": FluxPipeline,
+         "cache_dir" : cache_dir
+    },
         "AuraFlow": {
         "repo_id": "fal/AuraFlow",
         "pipeline_class": AuraFlowPipeline,
@@ -45,6 +50,7 @@ def generate_image_with_progress(pipe, prompt, num_steps, guidance_scale=None, s
         image = pipe(
             prompt,
             num_inference_steps=num_steps,
+            generator=generator,
             guidance_scale=guidance_scale,
             callback_on_step_end=callback,
         ).images[0]
@@ -54,6 +60,7 @@ def generate_image_with_progress(pipe, prompt, num_steps, guidance_scale=None, s
             prompt,
             num_inference_steps=num_steps,
             guidance_scale=guidance_scale,
+            generator=generator,
         ).images[0]
     elif hasattr(pipe, "callback_on_step_end") and not hasattr(pipe, "guidance_scale"):
         print(" has callback_on_step_end and NO guidance_scale")
@@ -68,6 +75,7 @@ def generate_image_with_progress(pipe, prompt, num_steps, guidance_scale=None, s
         image = pipe(
             prompt,
             num_inference_steps=num_steps,
+            generator=generator,
         ).images[0]
 
     return image