Update app.py

app.py

@@ -1,4 +1,43 @@
 import gradio as gr
 from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
 
-print("yeahhh")
+model_name = "Kongfha/PhraAphaiManee-LM"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
+
+nlp = pipeline("text-generation",
+                    model=model,
+                    tokenizer=tokenizer)
+
+def generate(input_sentence, max_length=140, top_k=50, temperature=1.0):
+    generated_text = nlp(input_sentence, 
+                        max_length=max_length, 
+                        do_sample=True, 
+                        top_k=top_k, 
+                        temperature=temperature)
+    return generated_text[0]['generated_text']
+
+inputs = [
+    gr.inputs.Textbox(label="Input Sentence"),
+    gr.inputs.Number(default=140, label="Max Length"),
+    gr.inputs.Number(default=50, label="Top K"),
+    gr.inputs.Slider(minimum=0.1, maximum=2.0, default=1.0, label="Temperature", step=0.1),
+]
+
+outputs = gr.outputs.Textbox(label="Generated Text")
+
+examples = [
+    ["๏ เรือล่อง", 100, 20, 0.5],
+    ["๏ แม้นชีวี", 120, 30, 0.7],
+    ["๏ หากวันใด", 140, 50, 1.0],
+    ["๏ หากจำเป็น", 160, 70, 1.5]
+]
+
+iface = gr.Interface(
+  fn=generate, 
+  inputs=inputs,
+  outputs=outputs,
+  examples=examples
+)
+
+iface.launch()