iespsurmreqmer's picture
Commit sin el token
43a3eff
raw
history blame
603 Bytes
import gradio as gr
from PIL import Image
from transformers import pipeline
captioner = pipeline("image-to-text", model="Salesforce/blip-image-captioning-large")
def describir(imagen):
if imagen is None:
return "Sube una imagen."
img = Image.fromarray(imagen)
result = captioner(img)[0]["generated_text"]
return result
demo = gr.Interface(
fn=describir,
inputs=gr.Image(type="numpy"),
outputs=gr.Textbox(),
title="Accesibilidad con Transformers",
description="Sube una imagen y un modelo Transformer generará una descripción detallada."
)
demo.launch()