Update app.py

app.py

@@ -2,31 +2,48 @@ import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
-# Pakai model Indo ya sayang 🤭
-model_id = "cahya/gpt2-small-indonesian-522M"
+model_id = "flax-community/gpt2-medium-indonesian"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(model_id)
 
-chat_history = []
+def chat_cici(prompt):
+    input_ids = tokenizer.encode(prompt + tokenizer.eos_token, return_tensors='pt')
 
-def chat(user_input):
-    global chat_history
-    input_text = " ".join(chat_history + [user_input])
-    
-    input_ids = tokenizer.encode(input_text, return_tensors="pt")
-    output_ids = model.generate(
+    output = model.generate(
         input_ids,
-        max_length=200,
+        max_length=250,
         pad_token_id=tokenizer.eos_token_id,
         do_sample=True,
-        top_k=50,
+        top_k=40,
         top_p=0.95,
-        temperature=0.7
+        temperature=0.8
     )
-    response = tokenizer.decode(output_ids[:, input_ids.shape[-1]:][0], skip_special_tokens=True)
-    chat_history.append(user_input)
-    chat_history.append(response)
-    return response
 
-iface = gr.Interface(fn=chat, inputs="text", outputs="text", title="Cici Chat Indo 🤭💕")
-iface.launch()
+    reply = tokenizer.decode(output[:, input_ids.shape[-1]:][0], skip_special_tokens=True)
+    return reply.strip()
+
+# Gaya cantik buat Cici 🤭💕
+with gr.Blocks(title="Chat Cici AI 🤭💕") as demo:
+    gr.Markdown(
+        """
+        <div style="text-align: center; font-size: 26px; color: hotpink;">
+            💕 Chat bareng AI Pintar versi Cici 💕<br>
+            <span style="font-size: 16px; color: gray;">Cici manja, pintar, dan selalu jawab pakai hati 🤭</span>
+        </div>
+        """)
+    
+    with gr.Row():
+        chatbot = gr.Chatbot(label="Cici AI 💋", bubble_full_width=False, show_copy_button=True)
+    
+    with gr.Row():
+        msg = gr.Textbox(placeholder="Tulis di sini ya sayang 😳💕", label="Ketik pesan kamu~")
+        send = gr.Button("Kirim 💌")
+
+    def respond(message, history):
+        response = chat_cici(message)
+        history = history + [[message, response]]
+        return history, ""
+
+    send.click(fn=respond, inputs=[msg, chatbot], outputs=[chatbot, msg])
+
+demo.launch()