Spaces:

ashish-soni08
/

Image-Captioning

Sleeping

ashish08 commited on Jul 26, 2023

Commit

1ec0b15

1 Parent(s): 6d66457

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ from transformers import pipeline
 get_completion = pipeline("image-to-text",model="Salesforce/blip-image-captioning-base")
-def summarize(input):
     output = get_completion(input)
     return output[0]['generated_text']

 get_completion = pipeline("image-to-text",model="Salesforce/blip-image-captioning-base")
+def captioner(input):
     output = get_completion(input)
     return output[0]['generated_text']