minakshi.mathpal commited on
Commit
c94f902
·
1 Parent(s): 0e01082

refactored app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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, # Pass the selected concept
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