bipin commited on
Commit
abd7bfe
1 Parent(s): 32613f0

added new example image

Browse files
Files changed (3) hide show
  1. app.py +2 -2
  2. dog_image.jpg +0 -0
  3. 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=True):
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=[["dog_image.jpg", "action"]],
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