Silvia Terragni commited on
Commit
ab4fce6
1 Parent(s): d16ed94

fixing caption suggestions

Browse files
Files changed (1) hide show
  1. text2image.py +10 -2
text2image.py CHANGED
@@ -130,11 +130,13 @@ def app():
130
  "Un gatto",
131
  "Due gatti",
132
  "Un fiore giallo",
133
- "Un gatto sopra una sedia",
 
 
134
  ]
135
  sugg_idx = -1
136
 
137
- col1, col2, col3, col4 = st.beta_columns([1, 1, 1, 2])
138
  with col1:
139
  if st.button(suggestions[0]):
140
  sugg_idx = 0
@@ -147,6 +149,12 @@ def app():
147
  with col4:
148
  if st.button(suggestions[3]):
149
  sugg_idx = 3
 
 
 
 
 
 
150
 
151
  col1, col2 = st.beta_columns([3, 1])
152
  with col1:
 
130
  "Un gatto",
131
  "Due gatti",
132
  "Un fiore giallo",
133
+ "Un fiore blu",
134
+ "Una coppia in montagna",
135
+ "Una coppia al tramonto"
136
  ]
137
  sugg_idx = -1
138
 
139
+ col1, col2, col3, col4, col5, col6 = st.beta_columns([1, 1, 1, 1, 1.5, 1.5])
140
  with col1:
141
  if st.button(suggestions[0]):
142
  sugg_idx = 0
 
149
  with col4:
150
  if st.button(suggestions[3]):
151
  sugg_idx = 3
152
+ with col5:
153
+ if st.button(suggestions[4]):
154
+ sugg_idx = 4
155
+ with col6:
156
+ if st.button(suggestions[5]):
157
+ sugg_idx = 5
158
 
159
  col1, col2 = st.beta_columns([3, 1])
160
  with col1: