Spaces:
Sleeping
Sleeping
File size: 677 Bytes
f35b47e | 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 transformers import pipeline
titulo = "Mi primer demo con Hugging Face"
descripcion = "Esta es mi primera demo con Hugging Face"
image_classifier = pipeline("image-classification", model="microsoft/swin-tiny-patch4-window7-224")
def classify_image(image):
results = image_classifier(image)
formatted_results = {result['label']: result['score'] for result in results}
return formatted_results
interface = gr.Interface(
fn=classify_image,
inputs=gr.Image(type="pil", label="Carga una imagen"),
outputs=gr.Label(num_top_classes=5),
title=titulo,
description=descripcion,
allow_flagging="never"
)
interface.launch() |