Spaces:
Runtime error
Runtime error
bipin
commited on
Commit
•
abd7bfe
1
Parent(s):
32613f0
added new example image
Browse files- app.py +2 -2
- dog_image.jpg +0 -0
- image.jpg +0 -0
app.py
CHANGED
@@ -4,7 +4,7 @@ from prefix_clip import download_pretrained_model, generate_caption
|
|
4 |
from gpt2_story_gen import generate_story
|
5 |
|
6 |
|
7 |
-
def main(pil_image, genre, model="Conceptual", use_beam_search=
|
8 |
model_file = "pretrained_weights.pt"
|
9 |
|
10 |
download_pretrained_model(model.lower(), file_to_save=model_file)
|
@@ -44,7 +44,7 @@ if __name__ == "__main__":
|
|
44 |
),
|
45 |
],
|
46 |
outputs=gr.outputs.Textbox(label="Generated story"),
|
47 |
-
examples=[["
|
48 |
enable_queue=True,
|
49 |
)
|
50 |
interface.launch()
|
|
|
4 |
from gpt2_story_gen import generate_story
|
5 |
|
6 |
|
7 |
+
def main(pil_image, genre, model="Conceptual", use_beam_search=False):
|
8 |
model_file = "pretrained_weights.pt"
|
9 |
|
10 |
download_pretrained_model(model.lower(), file_to_save=model_file)
|
|
|
44 |
),
|
45 |
],
|
46 |
outputs=gr.outputs.Textbox(label="Generated story"),
|
47 |
+
examples=[["image.jpg", "action"]],
|
48 |
enable_queue=True,
|
49 |
)
|
50 |
interface.launch()
|
dog_image.jpg
DELETED
Binary file (80.9 kB)
|
|
image.jpg
ADDED