Upload 5 files

app.py

@@ -43,7 +43,15 @@ def infer_core(image: Image.Image, text: str, max_new_tokens: int = 256, tempera
         out = model.generate(**inputs, max_new_tokens=int(max_new_tokens), temperature=float(temperature))
     return processor.decode(out[0], skip_special_tokens=True).strip()
 
-# ---------- UI ----------
+
+# ---------- Helper for API ----------
+def describe_raw(image: Image.Image, text: str = "Describe la imagen con detalle.",
+                 max_new_tokens: int = 256, temperature: float = 0.7) -> Dict[str, str]:
+    result = infer_core(image, text, max_new_tokens, temperature)
+    return {"text": result}
+
+
+# ---------- UI and API ----------
 with gr.Blocks(title="Salamandra Vision 7B · ZeroGPU") as demo:
     gr.Markdown("## Salamandra-Vision 7B · ZeroGPU\nImagen + texto → descripción.")
     with gr.Row():
@@ -55,17 +63,13 @@ with gr.Blocks(title="Salamandra Vision 7B · ZeroGPU") as demo:
             btn = gr.Button("Generar", variant="primary")
         with gr.Column():
             out = gr.Textbox(label="Descripción", lines=18)
+
+    # Endpoint for UI
     btn.click(infer_core, [in_img, in_txt, max_new, temp], out, api_name="describe")
 
-# ---------- API pura (sin UI) ----------
-# Exponemos un endpoint REST nítido (multipart/form-data o JSON base64) sin depender de componentes UI.
-# /api/describe_raw -> recibe {image,file} y campos simples.
-@gr.api()
-@spaces.GPU
-def describe_raw(image: gr.File, text: str = "Describe la imagen con detalle.", 
-                 max_new_tokens: int = 256, temperature: float = 0.7) -> Dict[str, str]:
-    img = Image.open(image)
-    result = infer_core(img, text, max_new_tokens, temperature)
-    return {"text": result}
+    # Endpoint for API (no UI)
+    demo.load(None, [gr.Image(label="image", type="pil"), gr.Textbox(value="Describe la imagen con detalle."),
+                     gr.Slider(16, 1024, value=256), gr.Slider(0.0, 1.5, value=0.7)],
+              describe_raw, api_name="describe_raw")
 
 demo.queue(concurrency_count=1, max_size=16).launch()

requirements.txt

@@ -1,5 +1,5 @@
 # app (ZeroGPU Gradio)
-gradio>=4.44.0
+gradio>=4.44.1
 spaces>=0.25.0
 transformers>=4.44.0
 torch>=2.2