Update app.py

app.py

@@ -1,39 +1,47 @@
 import gradio as gr
-from transformers import pipeline
+from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
 
 # --- CONFIGURATION ---
 
-MODEL_K2H = "ankitklakra/kurukh-to-hindi"
-MODEL_H2K = "ankitklakra/hindi-to-kurukh"
+MODEL_K2H_REPO = "ankitklakra/kurukh-to-hindi"
+MODEL_H2K_REPO = "ankitklakra/hindi-to-kurukh"
 
-# --- LOAD MODELS ---
+# --- LOAD RESOURCES ---
+# 1. Load the "Dictionary" (Tokenizer) from Google
 
+print("Loading Tokenizer...")
+tokenizer = AutoTokenizer.from_pretrained("google/mt5-small")
+
+# 2. Load the "Brains" (Your Custom Models)
 print("Loading Kurukh -> Hindi Model...")
-pipe_k2h = pipeline("text2text-generation", model=MODEL_K2H)
+model_k2h = AutoModelForSeq2SeqLM.from_pretrained(MODEL_K2H_REPO)
 
 print("Loading Hindi -> Kurukh Model...")
-pipe_h2k = pipeline("text2text-generation", model=MODEL_H2K)
+model_h2k = AutoModelForSeq2SeqLM.from_pretrained(MODEL_H2K_REPO)
+
+# 3. Create the Pipelines
+
+pipe_k2h = pipeline("text2text-generation", model=model_k2h, tokenizer=tokenizer)
+pipe_h2k = pipeline("text2text-generation", model=model_h2k, tokenizer=tokenizer)
 
 # --- TRANSLATION FUNCTION ---
 def translate_text(text, direction):
     if not text:
         return ""
     
-    # Select the correct brain based on user choice
+    # Select the correct pipeline
     if direction == "Kurukh -> Hindi":
         target_pipeline = pipe_k2h
     else:
         target_pipeline = pipe_h2k
     
     # Translate
-    # max_length=128 allow for longer sentences
     results = target_pipeline(text, max_length=128)
     return results[0]['generated_text']
 
 # --- THE USER INTERFACE ---
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     
-    # Header
     gr.Markdown(
         """
         # 🇮🇳 AI Kurukh (Kurux) Translator
@@ -42,7 +50,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         """
     )
     
-    # Input Section
     with gr.Row():
         direction = gr.Radio(
             ["Kurukh -> Hindi", "Hindi -> Kurukh"], 
@@ -63,10 +70,9 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             output_text = gr.Textbox(
                 label="Translation Result", 
                 lines=5, 
-                show_copy_button=True # Allows users to copy result
+                show_copy_button=True
             )
             
-    # Examples to help new users
     gr.Examples(
         examples=[
             ["निघै नामे इन्द्रा हिकै?", "Kurukh -> Hindi"],
@@ -78,7 +84,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         label="Click on an example to test:"
     )
 
-    # Logic Connection
     translate_btn.click(fn=translate_text, inputs=[input_text, direction], outputs=output_text)
 
 # Launch