Fix variable interpolation in app.py

app.py

@@ -3,12 +3,11 @@ import numpy as np
 from fastai.vision.all import load_learner, PILImage
 from huggingface_hub import hf_hub_download
 
-# Descargar y cargar export.pkl
-pkl = hf_hub_download(repo_id="${MODEL_REPO}", filename="export.pkl")
+# Aquí sí se expande MODEL_REPO
+pkl = hf_hub_download(repo_id="alramil/unet-segmentation-model", filename="export.pkl")
 learn = load_learner(pkl)
 learn.model.eval()
 
-# Paleta de colores para índices 0–4
 palette = np.array([
     [0,0,0],      # fondo
     [0,128,0],    # hojas
@@ -18,17 +17,15 @@ palette = np.array([
 ], dtype=np.uint8)
 
 def segment_image(img):
-    # FastAI internamente hace resize, tensor y normalize
     pred,_,_ = learn.predict(PILImage.create(img))
-    seg = np.array(pred)  # H×W array de 0–4
+    seg = np.array(pred)
     return palette[seg]
 
 demo = gr.Interface(
     fn=segment_image,
     inputs=gr.Image(type="pil", label="Sube imagen"),
-    outputs=gr.Image(type="numpy", label="Máscara segmentada"),
-    title="Segmentación de Racimos de Uva",
-    description="Modelo U-Net entrenado con FastAI"
+    outputs=gr.Image(type="numpy", label="Máscara"),
+    title="Segmentación Semántica de Racimos"
 )
 
 if __name__=="__main__":