CR7CAD commited on
Commit
b2cad31
·
verified ·
1 Parent(s): a084b90

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import os
6
  # function part
7
  # img2text
8
  def img2text(image_path):
9
- image_to_text = pipeline("image-to-text", model="dumperize/movie-picture-captioning")
10
  text = image_to_text(image_path)[0]["generated_text"]
11
  return text
12
 
 
6
  # function part
7
  # img2text
8
  def img2text(image_path):
9
+ image_to_text = pipeline("image-to-text", model="sooh-j/blip-image-captioning-base")
10
  text = image_to_text(image_path)[0]["generated_text"]
11
  return text
12