Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
4bd7b20
1
Parent(s):
0198045
Add FLUX.1-dev model attribution and improve project documentation
Browse files- Add proper attribution section with links to FLUX.1 [pro], license, blog, and model
- Enhance CLAUDE.md with detailed code architecture and key function references
🤖 Generated with [Claude Code](https://claude.ai/code)
Co-Authored-By: Claude <[email protected]>
CLAUDE.md
CHANGED
@@ -31,4 +31,17 @@ The app is automatically deployed to Hugging Face Spaces with ZeroGPU. Changes t
|
|
31 |
- Model runs in **bfloat16** precision for optimal performance on H200 GPUs
|
32 |
- Output resolution is **1024x1024** pixels
|
33 |
- Optimal parameters: 28 inference steps, guidance scale 3.5
|
34 |
-
- Prompt enhancement automatically adds Picasso-style descriptors
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
31 |
- Model runs in **bfloat16** precision for optimal performance on H200 GPUs
|
32 |
- Output resolution is **1024x1024** pixels
|
33 |
- Optimal parameters: 28 inference steps, guidance scale 3.5
|
34 |
+
- Prompt enhancement automatically adds Picasso-style descriptors
|
35 |
+
|
36 |
+
## Code Architecture
|
37 |
+
|
38 |
+
### Core Components
|
39 |
+
- **Model Loading**: `load_model()` function initializes FLUX.1-dev pipeline with bfloat16 precision
|
40 |
+
- **Image Generation**: `generate_image()` function handles prompt enhancement and image generation with configurable parameters
|
41 |
+
- **UI Interface**: Gradio Blocks layout with prompt input, parameter sliders, and example buttons
|
42 |
+
- **Example System**: Pre-defined example prompts accessible through vertical button layout
|
43 |
+
|
44 |
+
### Key Functions
|
45 |
+
- `load_model()` (app.py:9): Loads and configures the FLUX.1-dev pipeline
|
46 |
+
- `generate_image()` (app.py:27): Main generation function with Picasso-style prompt enhancement
|
47 |
+
- Example setter functions (app.py:88-97): Populate prompt input with predefined examples
|
app.py
CHANGED
@@ -115,6 +115,14 @@ with gr.Blocks(title="🎨 FLUX.1-dev-Picasso") as demo:
|
|
115 |
inputs=[prompt_input, steps_slider, guidance_slider],
|
116 |
outputs=[output_image, output_text]
|
117 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
118 |
|
119 |
if __name__ == "__main__":
|
120 |
demo.launch()
|
|
|
115 |
inputs=[prompt_input, steps_slider, guidance_slider],
|
116 |
outputs=[output_image, output_text]
|
117 |
)
|
118 |
+
|
119 |
+
# Attribution section
|
120 |
+
gr.Markdown("""
|
121 |
+
---
|
122 |
+
**Model Attribution:**
|
123 |
+
12B param rectified flow transformer guidance-distilled from [FLUX.1 [pro]](https://huggingface.co/black-forest-labs/FLUX.1-pro)
|
124 |
+
[[non-commercial license]](https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICENSE.md) [[blog]](https://blackforestlabs.ai/announcing-black-forest-labs/) [[model]](https://huggingface.co/black-forest-labs/FLUX.1-dev)
|
125 |
+
""")
|
126 |
|
127 |
if __name__ == "__main__":
|
128 |
demo.launch()
|