Add application file

app.py

@@ -1,7 +1,24 @@
 import gradio as gr
+from transformers import pipeline
 
-def greet(name):
-    return "Hello " + name + "!!"
+pipe = pipeline("text-generation", model="openchat/openchat_3.5")
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-demo.launch()
+# Prompt แบบ role-play
+prompt = """[INST] Extract structured data from the following message using this form:
+- วันที่ขึ้นของ
+- วันที่ส่งของ
+- ต้นทาง
+- ปลายทาง
+
+Message:
+เข้ารับของวัน พุธ  ที่  28/05/2568
+ส่งโหลด วัน พฤหัส  ที่ 29/05/2568 
+ห้วยกระเจา - BBL
+เข้ารับของวัน พฤหัส  ที่  29/05/2568
+ส่งโหลด วัน ศุกร์  ที่ 30/05/2568 
+ห้วยกระเจา - BBL
+[/INST]
+"""
+
+result = pipe(prompt, max_new_tokens=300, do_sample=False)
+print(result[0]["generated_text"])