minakshi.mathpal commited on
Commit ·
c94f902
1
Parent(s): 0e01082
refactored app.py
Browse files
app.py
CHANGED
|
@@ -151,7 +151,7 @@ if color_button:
|
|
| 151 |
config=st.session_state.config,
|
| 152 |
image_processor=st.session_state.image_processor,
|
| 153 |
prompt=prompt,
|
| 154 |
-
concepts=concept_name
|
| 155 |
output_dir="concept_images",
|
| 156 |
blue_loss_scale=0,
|
| 157 |
yellow_loss_scale=yellow_strength
|
|
|
|
| 151 |
config=st.session_state.config,
|
| 152 |
image_processor=st.session_state.image_processor,
|
| 153 |
prompt=prompt,
|
| 154 |
+
concepts=[concept_name] if concept_name else [],
|
| 155 |
output_dir="concept_images",
|
| 156 |
blue_loss_scale=0,
|
| 157 |
yellow_loss_scale=yellow_strength
|