Update app.py

app.py

@@ -1,8 +1,7 @@
 import gradio as gr
-from transformers import AutoTokenizer, AutoModelForCausalLM
+from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
-# Load GPT-2 Bahasa Indonesia (contoh: IndoGPT2 dari cahya)
 model_name = "cahya/gpt2-small-indonesian-522M"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
@@ -12,22 +11,19 @@ def generate_response(prompt):
     with torch.no_grad():
         output = model.generate(
             input_ids,
-            max_length=100,
+            max_length=80,
             do_sample=True,
             top_k=50,
             top_p=0.95,
             temperature=0.9,
-            pad_token_id=tokenizer.eos_token_id,
+            pad_token_id=tokenizer.eos_token_id
         )
     return tokenizer.decode(output[0], skip_special_tokens=True)
 
-# Gradio UI
-iface = gr.Interface(
+gr.Interface(
     fn=generate_response,
-    inputs=gr.Textbox(lines=4, placeholder="Ketik sesuatu yang nakal... 🤭"),
+    inputs=gr.Textbox(lines=4, placeholder="Tulis rayuanmu di sini... 🤭"),
     outputs="text",
     title="🧠 AI Nakal Bahasa Indonesia",
-    description="Powered by GPT-2 Indo | Coba ketik rayuan atau pertanyaan nakal 😳💕"
-)
-
-iface.launch()
+    description="Powered by GPT-2 Indo — Coba tanya yang manja, romantis, atau... 🌚"
+).launch()