working version flux

app.py

@@ -14,6 +14,7 @@ from app_qwen import demo as demo_qwen
 from app_sambanova import demo as demo_sambanova
 from app_together import demo as demo_together
 from app_xai import demo as demo_grok
+from app_flux import demo as demo_flux
 
 with gr.Blocks(fill_height=True) as demo:
     with gr.Tab("Meta Llama"):
@@ -47,6 +48,8 @@ with gr.Blocks(fill_height=True) as demo:
         demo_together.render()
     with gr.Tab("NVIDIA"):
         demo_nvidia.render()
+    with gr.Tab("Flux"):
+        demo_flux.render()
 
 if __name__ == "__main__":
     demo.launch(ssr_mode=False)

app_flux.py

@@ -0,0 +1,17 @@
+import os
+import replicate_gradio
+from utils import get_app
+
+demo = get_app(
+    models=[
+        "black-forest-labs/flux-depth-pro",
+        "black-forest-labs/flux-canny-pro",
+        "black-forest-labs/flux-fill-pro",
+        "black-forest-labs/flux-depth-dev"
+    ],
+    default_model="black-forest-labs/flux-depth-pro",
+    src=replicate_gradio.registry,
+)
+
+if __name__ == "__main__":
+    demo.launch()

requirements.txt

@@ -321,3 +321,4 @@ websockets==12.0
     # via gradio-client
 xai-gradio==0.0.2
     # via anychat (pyproject.toml)
+replicate-gradio @ git+https://github.com/AK391/replicate-gradio.git