Create app.py

app.py

@@ -0,0 +1,32 @@
+import gradio as gr
+from transformers import AutoModelForCausalLM, AutoTokenizer
+import torch
+
+# Pakai model Indo ya sayang 🤭
+model_id = "cahya/gpt2-small-indonesian-522M"
+tokenizer = AutoTokenizer.from_pretrained(model_id)
+model = AutoModelForCausalLM.from_pretrained(model_id)
+
+chat_history = []
+
+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(
+        input_ids,
+        max_length=200,
+        pad_token_id=tokenizer.eos_token_id,
+        do_sample=True,
+        top_k=50,
+        top_p=0.95,
+        temperature=0.7
+    )
+    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()