Hugging Face's logo

Spaces:
Aanisha
/
Image_to_story
Runtime error

App Files Community
Aanisha commited on
Commit
d1c036f
•
1 Parent(s): 2d49faa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,8 +8,8 @@ import re
8
 
9
 
10
 
11
- description = "<h4>Just upload an image, and generate a story for the image! GPT-2 is not perfect but it's fun to play with.</h4>"
12
- title = "<h1>Story generator from images using ViT and GPT2</h1>"
13
 
14
 
15
  model = VisionEncoderDecoderModel.from_pretrained("gagan3012/ViTGPT2_vizwiz").to('cpu')
8
 
9
 
10
 
11
+ description = "Just upload an image, and generate a short story for the image.\n PS: GPT-2 is not perfect but it's fun to play with.May take a minute for the output to generate. Enjoyy!!!"
12
+ title = "Story generator from images using ViT and GPT2"
13
 
14
 
15
  model = VisionEncoderDecoderModel.from_pretrained("gagan3012/ViTGPT2_vizwiz").to('cpu')