File size: 603 Bytes
43a3eff
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
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()