Aanisha commited on
Commit
2d49faa
1 Parent(s): dbbb340

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 = "Just upload an image, and generate a story for the image! GPT-2 ins't perfect but it's fun to play with."
12
- title = "Story generator from images using ViT and GPT2"
13
 
14
 
15
  model = VisionEncoderDecoderModel.from_pretrained("gagan3012/ViTGPT2_vizwiz").to('cpu')
 
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')