Update app.py

update space for two models

app.py

@@ -1,35 +1,36 @@
+# app.py
 import gradio as gr
 from functools import lru_cache
 from transformers import pipeline
 
-MODEL_ID = "hasmar03/mt5_id2md"
+# Ganti sesuai nama repo-mu di HF
+ID2MD = "hasmar03/mt5_id2md"
+MD2ID = "hasmar03/mt5_md2id"   # <- model baru hasil training md→id
 
-@lru_cache(maxsize=1)
-def get_pipe():
+@lru_cache(maxsize=2)
+def get_pipe(model_id: str):
     """
-    Cache pipeline supaya model hanya dimuat 1x.
-    Pakai CPU (device=-1) agar gratisan di Spaces jalan.
+    Cache per-model supaya load cuma sekali (jalan di CPU Spaces).
     """
     return pipeline(
         task="text2text-generation",
-        model=MODEL_ID,
-        tokenizer=MODEL_ID,
-        device=-1,                 # CPU
-        # framework & dtype otomatis oleh HF/Transformers
+        model=model_id,
+        tokenizer=model_id,
+        device=-1,   # CPU (gratis)
     )
 
 def translate(direction: str, text: str, max_new_tokens: int = 64) -> str:
-    if not text or not text.strip():
+    text = (text or "").strip()
+    if not text:
         return ""
 
-    # Prefix sesuai format training mT5 kamu
+    # pilih model berdasar arah
+    model_id = ID2MD if direction == "id2md" else MD2ID
+
+    # prefix sesuai format training
     prompt = f"translate {direction}: {text}"
 
-    # Setting decoding yang umum untuk terjemahan
-    # - num_beams memperbaiki kualitas (sedikit lebih lambat tapi OK di CPU kecil)
-    # - no_repeat_ngram_size mengurangi pengulangan
-    # - early_stopping untuk hentikan beam kalau sudah cukup baik
-    out = get_pipe()(
+    out = get_pipe(model_id)(
         prompt,
         max_new_tokens=int(max_new_tokens),
         num_beams=5,
@@ -37,7 +38,6 @@ def translate(direction: str, text: str, max_new_tokens: int = 64) -> str:
         no_repeat_ngram_size=3,
         early_stopping=True,
     )[0]["generated_text"]
-
     return out
 
 with gr.Blocks(title="mT5 id↔md Translator (HF Space API)") as demo:
@@ -63,16 +63,13 @@ with gr.Blocks(title="mT5 id↔md Translator (HF Space API)") as demo:
         label="Contoh cepat",
     )
 
-# ==== Queue versi-agnostik (jalan di Gradio 4.x & 5.x) ====
+# kompatibel Gradio 4.x/5.x
 try:
-    # Gradio 5.x (beberapa build menolak argumen apa pun)
     demo.queue()
 except TypeError:
     try:
-        # Gradio 5.x lain (punya max_size)
         demo.queue(max_size=12)
     except TypeError:
-        # Gradio 4.x (pakai concurrency_count)
         demo.queue(concurrency_count=1, max_size=12)
 
 if __name__ == "__main__":