k3ybladewielder commited on
Commit
a22b0d3
1 Parent(s): 5ffa3e1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,8 +10,8 @@ import warnings
10
  warnings.filterwarnings("ignore")
11
 
12
  task = "image-to-text"
13
- model = "Salesforce/blip-image-captioning-base"
14
- image_captioner = pipeline("image-to-text", model = "Salesforce/blip-image-captioning-large")
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)