Update app.py

app.py

@@ -29,7 +29,7 @@ def infer(message, history, enable_history=False, max_new_tokens=512):
         )
         model.eval()
 
-    # Optional history
+    # Add history only if enabled
     if enable_history and history:
         prev = "\n".join(
             [f"User: {h[0]}\nGamunu: {h[1]}" for h in history if h[1] is not None]
@@ -63,7 +63,7 @@ with gr.Blocks() as demo:
         enable_history = gr.Checkbox(label="Enable chat history", value=False)
         max_new_tokens = gr.Slider(64, 1024, value=512, step=32, label="🔢 Max New Tokens")
 
-    # Chat interface with working title
+    # Use only stable arguments (compatible everywhere)
     chat = gr.ChatInterface(
         fn=lambda message, history: infer(message, history, enable_history.value, max_new_tokens.value),
         title="Gamunu 4B Instruct Alpha",
@@ -73,10 +73,7 @@ with gr.Blocks() as demo:
             ["රු. 250 කින් මිලදී ගත් දේ රු. 400 නම් එය ප්‍රතිශතයකින් කීයද වැඩි වී ඇත?"],
             ["ඔබ ගුරුවරයෙකු ලෙස ක්‍රියාකරන්න. ශිෂ්‍යයාට ඉතිහාසය උගන්වන්න."],
             ["ඔබ කවියෙකු ලෙස 'ගඟක්' ගැන කෙටි කවියක් ලියන්න."],
-        ],
-        retry_btn=None,
-        undo_btn=None,
-        clear_btn="🧹 Clear Chat",
+        ]
     )
 
     gr.Markdown("""