Update app.py

app.py

@@ -1,21 +1,31 @@
+import os
 import gradio as gr
 from transformers import pipeline
 
-# Load your model from Hugging Face Hub
+# Model ID
 model_id = "Badhon/Bangla_punctuation_restore"
-punctuator = pipeline("text2text-generation", model=model_id)
 
-# Function to restore punctuation
+# Load model with HF token (required if model is private)
+punctuator = pipeline(
+    "text2text-generation",
+    model=model_id,
+    token=os.getenv("HF_TOKEN")
+)
+
+# Restore punctuation
 def restore_punctuation(text):
     if not text:
         return ""
-    outputs = punctuator(text)
-    return outputs[0]["generated_text"]
+    result = punctuator(text, max_length=256)
+    return result[0]["generated_text"]
 
 # Gradio UI
 demo = gr.Interface(
     fn=restore_punctuation,
-    inputs=gr.Textbox(lines=4, placeholder="Enter unpunctuated Bangla text..."),
+    inputs=gr.Textbox(
+        lines=4,
+        placeholder="Enter unpunctuated Bangla text..."
+    ),
     outputs="text",
     title="Bangla Punctuation Restoration",
     description="Type Bangla text without punctuation and get back text with punctuation restored!"