Spaces:
Sleeping
Sleeping
k3ybladewielder
commited on
Commit
•
a22b0d3
1
Parent(s):
5ffa3e1
Update app.py
Browse files
app.py
CHANGED
@@ -10,8 +10,8 @@ import warnings
|
|
10 |
warnings.filterwarnings("ignore")
|
11 |
|
12 |
task = "image-to-text"
|
13 |
-
model = "
|
14 |
-
image_captioner = pipeline("image-to-text", model =
|
15 |
|
16 |
def captioner(image):
|
17 |
result = image_captioner(image)
|
|
|
10 |
warnings.filterwarnings("ignore")
|
11 |
|
12 |
task = "image-to-text"
|
13 |
+
model = "nlpconnect/vit-gpt2-image-captioning"
|
14 |
+
image_captioner = pipeline("image-to-text", model = model)
|
15 |
|
16 |
def captioner(image):
|
17 |
result = image_captioner(image)
|