Add title

app.py

@@ -38,8 +38,10 @@ def respond(
 
     outputs = pipeline(
         messages,
-        max_new_tokens=256,
-        eos_token_id=terminators,
+        max_new_tokens=max_tokens,
+        temperature=temprature,
+        top_p=top_p,
+        eos_token_id=terminators
     )
     
     yield outputs[0]["generated_text"][-1]["content"]
@@ -47,21 +49,22 @@ def respond(
 """
 For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
 """
-demo = gr.ChatInterface(
-    respond,
-    additional_inputs=[
-        gr.Textbox(value="Kamu adalah seorang asisten yang baik", label="System message"),
-        gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
-        gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
-        gr.Slider(
-            minimum=0.1,
-            maximum=1.0,
-            value=0.95,
-            step=0.05,
-            label="Top-p (nucleus sampling)",
-        ),
-    ],
-)
+with gr.Blocks() as demo:
+    gr.Markdown("# 🇮🇩 Sahabat AI (Gemma)")
+    gr.ChatInterface(
+        respond,
+        additional_inputs=[
+            gr.Slider(minimum=1, maximum=2048, value=256, step=1, label="Max new tokens"),
+            gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
+            gr.Slider(
+                minimum=0.1,
+                maximum=1.0,
+                value=0.95,
+                step=0.05,
+                label="Top-p (nucleus sampling)",
+            ),
+        ],
+    )
 
 
 if __name__ == "__main__":