hra commited on
Commit
2170c7e
1 Parent(s): e49fa15

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -6
app.py CHANGED
@@ -74,7 +74,8 @@ def infer(prompt,
74
  greedy_decoding = False,
75
  return_full_text = False):
76
 
77
-
 
78
  top_k = None if top_k == 0 else top_k
79
  do_sample = False if num_beams > 0 else not greedy_decoding
80
  num_beams = None if (greedy_decoding or num_beams == 0) else num_beams
@@ -148,7 +149,7 @@ def getadvertisement(topic):
148
  else:
149
  input_keyword=getadline(random.choice('abcdefghijklmnopqrstuvwxyz'))
150
  if 'Unsafe content found' in input_keyword:
151
- input_keyword='Abstarct art with splash of colors'
152
  prompt_SD=input_keyword+','+prompt_image
153
  # generate image
154
  image = generate_image(prompt_SD)
@@ -164,10 +165,12 @@ with gr.Blocks() as demo:
164
  gr.Markdown(
165
  """Enter a prompt and get the tee shirt design. Use examples as a guide. We use an equally powerful AI model bigscience/bloom."""
166
  )
167
- textbox = gr.Textbox(placeholder="Enter prompt...", lines=1,label='Your prompt')
168
- btn = gr.Button("Generate")
169
- #output1 = gr.Textbox(lines=2,label='Market Sizing Framework')
170
- output_image = gr.components.Image(label="Your tee shirt")
 
 
171
 
172
 
173
  btn.click(getadvertisement,inputs=[textbox], outputs=[output_image])
 
74
  greedy_decoding = False,
75
  return_full_text = False):
76
 
77
+ print('Empty input')
78
+ print(prompt)
79
  top_k = None if top_k == 0 else top_k
80
  do_sample = False if num_beams > 0 else not greedy_decoding
81
  num_beams = None if (greedy_decoding or num_beams == 0) else num_beams
 
149
  else:
150
  input_keyword=getadline(random.choice('abcdefghijklmnopqrstuvwxyz'))
151
  if 'Unsafe content found' in input_keyword:
152
+ input_keyword='Abstract art with a splash of colors'
153
  prompt_SD=input_keyword+','+prompt_image
154
  # generate image
155
  image = generate_image(prompt_SD)
 
165
  gr.Markdown(
166
  """Enter a prompt and get the tee shirt design. Use examples as a guide. We use an equally powerful AI model bigscience/bloom."""
167
  )
168
+ with gr.Row() as row:
169
+ with gr.Column():
170
+ textbox = gr.Textbox(placeholder="Enter prompt...", lines=1,label='Your prompt')
171
+ btn = gr.Button("Generate")
172
+ with gr.Column():
173
+ output_image = gr.components.Image(label="Your tee shirt")
174
 
175
 
176
  btn.click(getadvertisement,inputs=[textbox], outputs=[output_image])