Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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
|
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')
|