Update app.py

app.py

@@ -5,24 +5,26 @@ from PIL import Image
 import base64
 import io
 
-# Modelo
+
 processor = AutoImageProcessor.from_pretrained("facebook/dinov2-base")
 model = AutoModel.from_pretrained("facebook/dinov2-base")
 
-# Recibe string base64
 def get_embedding(base64_str):
-    # Separar encabezado
     header, encoded = base64_str.split(",", 1)
-    # Decodificar
-    image_bytes = base64.b64decode(encoded)
-    image = Image.open(io.BytesIO(image_bytes)).convert("RGB")
+    image_data = base64.b64decode(encoded)
+    image = Image.open(io.BytesIO(image_data)).convert("RGB")
 
-    # Procesar
     inputs = processor(images=image, return_tensors="pt")
     with torch.no_grad():
         embeddings = model(**inputs).last_hidden_state[:, 0]
     return embeddings.squeeze().tolist()
 
-# Gradio Interface
-iface.queue()
-gr.Interface(fn=get_embedding, inputs="text", outputs="json").launch()
+
+iface = gr.Interface(
+    fn=get_embedding,
+    inputs="text",  # ahora recibimos un string base64
+    outputs="json",
+    description="Microservicio para extraer embeddings desde base64."
+)
+
+iface.queue()  # 👈 Esta línea activa el sistema de event_id y polling