Spaces:
Runtime error
Runtime error
najoungkim
commited on
Commit
•
8d5ee19
1
Parent(s):
fe96438
Added Done print statement at end of inference loop.
Browse files
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 |
|