najoungkim commited on
Commit
8d5ee19
1 Parent(s): fe96438

Added Done print statement at end of inference loop.

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -175,6 +175,7 @@ def run_inference(prompt, num_roundtrips=3, num_images=1):
175
  outputs.append(output_caption)
176
  prompt = caption
177
 
 
178
  return outputs
179
 
180
 
@@ -188,7 +189,7 @@ for _ in range(int(num_roundtrips)):
188
  outputs.append(gr.outputs.HTML(label=""))
189
 
190
  description = """
191
- Round trip DALL·E-mini iterates between DALL·E generation and image captioning, inspired by round trip translation!
192
  """
193
  article = "<p style='text-align: center'>Put together by: Najoung Kim | Dall-E Mini code from flax-community/dalle-mini | Caption code from SRDdev/Image-Caption</p>"
194
 
 
175
  outputs.append(output_caption)
176
  prompt = caption
177
 
178
+ print("Done.")
179
  return outputs
180
 
181
 
 
189
  outputs.append(gr.outputs.HTML(label=""))
190
 
191
  description = """
192
+ Round trip DALL·E-mini iterates between DALL·E generation and image captioning, inspired by round trip translation! FYI: it takes forever because the app is running on CPU.
193
  """
194
  article = "<p style='text-align: center'>Put together by: Najoung Kim | Dall-E Mini code from flax-community/dalle-mini | Caption code from SRDdev/Image-Caption</p>"
195