abidlabs HF staff commited on
Commit
37ca165
1 Parent(s): 7e92753

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -6,10 +6,9 @@ p = pipeline("text-generation", "EleutherAI/gpt-neo-1.3B")
6
 
7
  im2caption = gr.Interface.load("spaces/akhaliq/CLIP_prefix_captioning")
8
 
9
- im2caption = gr.Interface.load("spaces/akhaliq/CLIP_prefix_captioning")
10
-
11
  def im2story(img):
12
  caption = im2caption(img, "COCO")
 
13
  story = p("Let me tell you a story about " + caption)[0]["generated_text"]
14
  return story
15
 
 
6
 
7
  im2caption = gr.Interface.load("spaces/akhaliq/CLIP_prefix_captioning")
8
 
 
 
9
  def im2story(img):
10
  caption = im2caption(img, "COCO")
11
+ caption = caption[0].lower() + caption[1:]
12
  story = p("Let me tell you a story about " + caption)[0]["generated_text"]
13
  return story
14