Update Llama 3.1 8B robot planning space with improvements

README.md

@@ -4,7 +4,7 @@ emoji: 🤖
 colorFrom: blue
 colorTo: green
 sdk: gradio
-sdk_version: 3.50.2
+sdk_version: 4.32.2
 app_file: app.py
 pinned: false
 license: llama3.1
@@ -68,7 +68,7 @@ Convert natural language commands into structured task sequences for constructio
 - **Architecture**: Llama 3.1 8B + QLoRA adapters
 - **Quantization**: 4-bit (NF4) with double quantization
 - **Framework**: Transformers + PEFT + BitsAndBytesConfig
-- **Interface**: Gradio 3.50.2 (stable version)
+- **Interface**: Gradio 4.32.2 (stable version)
 - **Hardware**: T4-MEDIUM (16GB VRAM)
 
 ## ⚡ Performance Notes

app.py

@@ -1,3 +1,4 @@
+
 import gradio as gr
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
@@ -135,21 +136,24 @@ def chat_interface(message, history, max_tokens, temperature, top_p):
     
     try:
         response = generate_response(message, max_tokens, temperature, top_p)
-        history.append([message, response])  # 使用列表而不是元组，兼容 Gradio 3.x
+        history.append((message, response))
         return history, ""
     except Exception as chat_error:
         error_msg = f"❌ Chat Error: {str(chat_error)}"
-        history.append([message, error_msg])
+        history.append((message, error_msg))
         return history, ""
 
-# 创建 Gradio 应用 - 简化配置避免 bug
+# 创建 Gradio 应用
 with gr.Blocks(
-    title="Robot Task Planning - Llama 3.1 8B",
-    theme="default",  # 使用默认主题
-    css=".gradio-container {max-width: 1200px; margin: auto;}"
+    title="Robot Task Planning - Llama 3.1 8B", 
+    theme=gr.themes.Soft(),
+    css="""
+    .gradio-container {
+        max-width: 1200px;
+        margin: auto;
+    }
+    """
 ) as demo:
-    
-    # 标题和说明
     gr.Markdown("""
     # 🤖 Llama 3.1 8B - Robot Task Planning
     
@@ -164,35 +168,38 @@ with gr.Blocks(
     
     with gr.Row():
         with gr.Column(scale=3):
-            # 聊天界面 - 简化参数
             chatbot = gr.Chatbot(
                 label="Task Planning Results",
-                height=500
+                height=500,
+                show_label=True,
+                container=True,
+                bubble_full_width=False,
+                show_copy_button=True
             )
             
-            # 输入框 - 简化参数
             msg = gr.Textbox(
                 label="Robot Command",
                 placeholder="Enter robot task command (e.g., 'Deploy Excavator 1 to Soil Area 1')...",
                 lines=2,
-                max_lines=5
+                max_lines=5,
+                show_label=True,
+                container=True
             )
             
-            # 按钮
             with gr.Row():
-                send_btn = gr.Button("🚀 Generate Tasks", variant="primary")
-                clear_btn = gr.Button("🗑️ Clear")
+                send_btn = gr.Button("🚀 Generate Tasks", variant="primary", size="sm")
+                clear_btn = gr.Button("🗑️ Clear", variant="secondary", size="sm")
         
         with gr.Column(scale=1):
             gr.Markdown("### ⚙️ Generation Settings")
             
-            # 参数控制 - 简化配置
             max_tokens = gr.Slider(
                 minimum=50,
                 maximum=500,
                 value=200,
                 step=10,
-                label="Max Tokens"
+                label="Max Tokens",
+                info="Maximum number of tokens to generate"
             )
             
             temperature = gr.Slider(
@@ -200,7 +207,8 @@ with gr.Blocks(
                 maximum=2.0,
                 value=0.7,
                 step=0.1,
-                label="Temperature"
+                label="Temperature",
+                info="Controls randomness (lower = more focused)"
             )
             
             top_p = gr.Slider(
@@ -208,7 +216,8 @@ with gr.Blocks(
                 maximum=1.0,
                 value=0.9,
                 step=0.05,
-                label="Top-p"
+                label="Top-p",
+                info="Nucleus sampling threshold"
             )
             
             gr.Markdown("""
@@ -217,29 +226,28 @@ with gr.Blocks(
             Loading time: ~3-5 minutes
             """)
     
-    # 示例 - 简化格式
-    examples_data = ['Deploy Excavator 1 to Soil Area 1 for excavation.', 'Send Dump Truck 1 to collect material from Excavator 1, then unload at storage area.', 'Move all robots to avoid Puddle 1 after inspection.', 'Deploy multiple excavators to different soil areas simultaneously.', 'Coordinate dump trucks to transport materials from excavation site to storage.', 'Send robot to inspect rock area, then avoid with all other robots if dangerous.', 'Return all robots to start position after completing tasks.', 'Create a sequence: excavate, load, transport, unload, repeat.']
+    # 示例对话
     gr.Examples(
-        examples=examples_data,
+        examples=['Deploy Excavator 1 to Soil Area 1 for excavation.', 'Send Dump Truck 1 to collect material from Excavator 1, then unload at storage area.', 'Move all robots to avoid Puddle 1 after inspection.', 'Deploy multiple excavators to different soil areas simultaneously.', 'Coordinate dump trucks to transport materials from excavation site to storage.', 'Send robot to inspect rock area, then avoid with all other robots if dangerous.', 'Return all robots to start position after completing tasks.', 'Create a sequence: excavate, load, transport, unload, repeat.'],
         inputs=msg,
         label="💡 Example Robot Commands"
     )
     
-    # 事件绑定 - Gradio 3.x 风格
+    # 事件处理
     msg.submit(
-        fn=chat_interface,
+        chat_interface,
         inputs=[msg, chatbot, max_tokens, temperature, top_p],
         outputs=[chatbot, msg]
     )
     
     send_btn.click(
-        fn=chat_interface,
+        chat_interface,
         inputs=[msg, chatbot, max_tokens, temperature, top_p],
         outputs=[chatbot, msg]
     )
     
     clear_btn.click(
-        fn=lambda: ([], ""),
+        lambda: ([], ""),
         outputs=[chatbot, msg]
     )
 

requirements.txt

@@ -1,5 +1,4 @@
-gradio==3.50.2
-pydantic==1.10.12
+gradio==4.32.2
 transformers==4.44.2
 torch==2.1.0
 peft==0.7.1