Spaces:

k3ybladewielder
/

gen_app_imagecap

Sleeping

k3ybladewielder commited on Aug 9, 2023

Commit

5ffa3e1

1 Parent(s): 4493365

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ warnings.filterwarnings("ignore")
 task = "image-to-text"
 model = "Salesforce/blip-image-captioning-base"
-image_captioner = pipeline("image-to-text", model = "Salesforce/blip-image-captioning-base")
 def captioner(image):
     result = image_captioner(image)

 task = "image-to-text"
 model = "Salesforce/blip-image-captioning-base"
+image_captioner = pipeline("image-to-text", model = "Salesforce/blip-image-captioning-large")
 def captioner(image):
     result = image_captioner(image)