File size: 692 Bytes
761cd02 a2142f7 4b4c90a 761cd02 4b4c90a 761cd02 a2142f7 761cd02 4b4c90a 761cd02 5342f17 761cd02 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
import gradio as gr
from transformers import pipeline
from PIL import Image
# Initialize the pipeline with the image captioning model
caption_pipeline = pipeline("image-to-text", model="nlpconnect/vit-gpt2-image-captioning")
def generate_caption(image):
# The image is received as a PIL Image, so no need for conversion
result = caption_pipeline(image)
caption = result[0]["generated_text"]
return caption
# Setup the Gradio interface
interface = gr.Interface(fn=generate_caption,
inputs=gr.components.Image(type="pil", label="Upload an Image"),
outputs=gr.components.Textbox(label="Generated Caption"))
interface.launch()
|