luckyo87 commited on
Commit
d5a5ca6
·
1 Parent(s): c25dcdf

Add application file

Browse files
Files changed (1) hide show
  1. app.py +9 -2
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(fn=classify_image, inputs=gr.Image(type="pil"), outputs="text")
 
 
 
 
 
 
 
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()