mkjaramillo commited on
Commit
3ee2c8e
·
1 Parent(s): 832bcaa

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +30 -0
app.py ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import requests
3
+ import json
4
+
5
+ # Función para realizar la predicción
6
+ def classify_image(image):
7
+ # URL de la API de inferencia de Hugging Face
8
+ api_url = "https://api-inference.huggingface.co/models/nombre_del_usuario/nombre_del_modelo"
9
+
10
+ # Preparar la imagen para la API de inferencia
11
+ files = {'file': image}
12
+
13
+ # Realizar la solicitud de predicción a la API de inferencia
14
+ response = requests.post(api_url, files=files)
15
+
16
+ # Obtener el resultado de la predicción
17
+ result = json.loads(response.content)
18
+
19
+ # Retornar el resultado de la predicción
20
+ return result["label"]
21
+
22
+ # Configurar la interfaz de Gradio
23
+ iface = gr.Interface(fn=classify_image,
24
+ inputs="image",
25
+ outputs="text",
26
+ capture_session=True,
27
+ title="Clasificador de Imágenes")
28
+
29
+ # Ejecutar la interfaz de Gradio
30
+ iface.launch()