Spaces:
Running
Running
Add application file
Browse files
app.py
CHANGED
@@ -16,6 +16,13 @@ def classify_image(image):
|
|
16 |
predicted_class = model.config.id2label[predicted_class_idx]
|
17 |
return predicted_class
|
18 |
|
19 |
-
# Crear la interfaz de Gradio
|
20 |
-
interface = gr.Interface(
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
21 |
interface.launch()
|
|
|
16 |
predicted_class = model.config.id2label[predicted_class_idx]
|
17 |
return predicted_class
|
18 |
|
19 |
+
# Crear la interfaz de Gradio con título y descripción
|
20 |
+
interface = gr.Interface(
|
21 |
+
fn=classify_image,
|
22 |
+
inputs=gr.Image(type="pil"),
|
23 |
+
outputs="text",
|
24 |
+
title="Clasificador de Gatos vs Perros",
|
25 |
+
description="Sube una imagen de un gato o un perro y este modelo basado en ViT te dirá qué animal aparece en la imagen."
|
26 |
+
)
|
27 |
+
|
28 |
interface.launch()
|