update

app.py

@@ -5,7 +5,6 @@ from generate_poem import generate_poem_interface
 gr_interface = gr.TabbedInterface(
     [complete_poem_interface, generate_poem_interface],
     title="Thơ AI",
-    description="Ứng dụng mô hình ngôn ngữ cực lớn để sinh ra các bài thơ Việt Nam",
 )
 
 gr_interface.launch()

complete_poem.py

@@ -4,16 +4,16 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 tokenizer = AutoTokenizer.from_pretrained("vinai/phobert-base")
 # Define your models
 models = {
-    "Luc Bat": AutoModelForCausalLM.from_pretrained(
+    "Lục Bát": AutoModelForCausalLM.from_pretrained(
         "Libosa2707/vietnamese-poem-luc-bat-gpt2"
     ),
-    "Bay Chu": AutoModelForCausalLM.from_pretrained(
+    "Bảy Chữ": AutoModelForCausalLM.from_pretrained(
         "Libosa2707/vietnamese-poem-bay-chu-gpt2"
     ),
-    "Tam Chu": AutoModelForCausalLM.from_pretrained(
+    "Tám Chữ": AutoModelForCausalLM.from_pretrained(
         "Libosa2707/vietnamese-poem-tam-chu-gpt2"
     ),
-    "Nam Chu": AutoModelForCausalLM.from_pretrained(
+    "Năm Chữ": AutoModelForCausalLM.from_pretrained(
         "Libosa2707/vietnamese-poem-nam-chu-gpt2"
     ),
 }
@@ -56,9 +56,17 @@ def complete_poem(text, style):
 complete_poem_interface = gr.Interface(
     fn=complete_poem,
     inputs=[
-        gr.components.Textbox(lines=1, placeholder="First words of the poem"),
+        # Title and description
+        gr.Interface.Title("Viết tiếp áng thơ hay"),
+        gr.components.Textbox(
+            lines=1,
+            placeholder="Tôi đâu có biết làm thơ",
+            label="Những áng thơ đầu tiên",
+        ),
         gr.components.Dropdown(
-            choices=["Luc Bat", "Bay Chu", "Tam Chu", "Nam Chu"], label="Style"
+            choices=["Lục Bát", "Bảy Chữ", "Tám Chữ", "Năm Chữ"],
+            label="Kiểu thơ",
+            default="Lục Bát",
         ),
     ],
     outputs="text",