#Filter the warnings | |
import warnings | |
warnings.filterwarnings('ignore') | |
from transformers import logging | |
logging.set_verbosity_error() | |
#Import the necessary | |
import gradio as gr | |
from transformers import pipeline | |
#Perform Image Captioning | |
get_completion = pipeline("image-to-text",model="Salesforce/blip-image-captioning-base") | |
#Launch the input | |
def launch(input): | |
out=get_completion(input) | |
return out[0]['generated_text'] | |
#Define an interface | |
interface=gr.Interface(fn=launch,inputs=gr.Image(type="pil"),outputs="text",title="Image Captioning with BLIP") | |
interface.launch() | |