Mikiko Bazeley commited on
Commit
347bec2
1 Parent(s): 57eccf2

Finished troubleshooting

Browse files
Files changed (2) hide show
  1. .gitignore +1 -0
  2. pages/1_Generate_Holiday_Postcard.py +21 -1
.gitignore CHANGED
@@ -293,3 +293,4 @@ cython_debug/
293
  # and can be added to the global gitignore or merged into this file. For a more nuclear
294
  # option (not recommended) you can uncomment the following to ignore the entire idea folder.
295
  #.idea/
 
 
293
  # and can be added to the global gitignore or merged into this file. For a more nuclear
294
  # option (not recommended) you can uncomment the following to ignore the entire idea folder.
295
  #.idea/
296
+ .DS_Store
pages/1_Generate_Holiday_Postcard.py CHANGED
@@ -83,6 +83,19 @@ Choose from two FLUX models: whether you’re aiming for lightning speed or extr
83
 
84
  model_choice = st.selectbox("Select the model:", ["flux-1-schnell-fp8", "flux-1-dev-fp8"])
85
 
 
 
 
 
 
 
 
 
 
 
 
 
 
86
  # Dropdown to select an example message or write a custom one
87
  st.divider()
88
  st.subheader("3️⃣ Step 3: Craft the Perfect Message 🎁")
@@ -142,7 +155,14 @@ if generate_button:
142
  steps = 30 if model_choice == "flux-1-dev-fp8" else 4
143
 
144
  # Generate image using the helper utility
145
- generated_image = generate_flux_image(model_choice, prompt, steps, fireworks_api_key)
 
 
 
 
 
 
 
146
 
147
  # Get the selected font path from configuration
148
  font_path = fonts[font_choice]
 
83
 
84
  model_choice = st.selectbox("Select the model:", ["flux-1-schnell-fp8", "flux-1-dev-fp8"])
85
 
86
+ # Expose parameters like guidance_scale, inference steps, and seed after model selection
87
+ st.divider()
88
+ st.subheader("2️⃣A: Customize Your Model Parameters ⚙️")
89
+ st.markdown("""
90
+ Now that you've selected your model, fine-tune the parameters to adjust how the FLUX model generates your holiday card.
91
+ """)
92
+
93
+ # Parameter options exposed after model selection
94
+ guidance_scale = st.slider("Guidance Scale", min_value=0.0, max_value=20.0, value=7.5, step=0.1)
95
+ num_inference_steps = st.slider("Number of Inference Steps", min_value=1, max_value=100, value=50, step=1)
96
+ seed = st.slider("Random Seed", min_value=0, max_value=1000, value=42)
97
+
98
+
99
  # Dropdown to select an example message or write a custom one
100
  st.divider()
101
  st.subheader("3️⃣ Step 3: Craft the Perfect Message 🎁")
 
155
  steps = 30 if model_choice == "flux-1-dev-fp8" else 4
156
 
157
  # Generate image using the helper utility
158
+ generated_image = generate_flux_image(
159
+ model_path=model_choice, # Correctly pass the model choice
160
+ api_key=fireworks_api_key, # Fireworks API key
161
+ prompt=prompt, # User's prompt for generation
162
+ steps=steps, # Number of inference steps
163
+ guidance_scale=guidance_scale, # Guidance scale (optional if unchanged)
164
+ seed=seed # Random seed (optional if unchanged)
165
+ )
166
 
167
  # Get the selected font path from configuration
168
  font_path = fonts[font_choice]