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: 4.32.2
+sdk_version: 3.50.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 4.32.2 (stable version)
+- **Interface**: Gradio 3.50.2 (stable version)
 - **Hardware**: T4-MEDIUM (16GB VRAM)
 
 ## ⚡ Performance Notes

app.py

@@ -1,4 +1,3 @@
-
 import gradio as gr
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
@@ -136,24 +135,21 @@ def chat_interface(message, history, max_tokens, temperature, top_p):
     
     try:
         response = generate_response(message, max_tokens, temperature, top_p)
-        history.append((message, response))
+        history.append([message, response])  # 使用列表而不是元组，兼容 Gradio 3.x
         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 应用
+# 创建 Gradio 应用 - 简化配置避免 bug
 with gr.Blocks(
-    title="Robot Task Planning - Llama 3.1 8B", 
-    theme=gr.themes.Soft(),
-    css="""
-    .gradio-container {
-        max-width: 1200px;
-        margin: auto;
-    }
-    """
+    title="Robot Task Planning - Llama 3.1 8B",
+    theme="default",  # 使用默认主题
+    css=".gradio-container {max-width: 1200px; margin: auto;}"
 ) as demo:
+    
+    # 标题和说明
     gr.Markdown("""
     # 🤖 Llama 3.1 8B - Robot Task Planning
     
@@ -168,38 +164,35 @@ with gr.Blocks(
     
     with gr.Row():
         with gr.Column(scale=3):
+            # 聊天界面 - 简化参数
             chatbot = gr.Chatbot(
                 label="Task Planning Results",
-                height=500,
-                show_label=True,
-                container=True,
-                bubble_full_width=False,
-                show_copy_button=True
+                height=500
             )
             
+            # 输入框 - 简化参数
             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,
-                show_label=True,
-                container=True
+                max_lines=5
             )
             
+            # 按钮
             with gr.Row():
-                send_btn = gr.Button("🚀 Generate Tasks", variant="primary", size="sm")
-                clear_btn = gr.Button("🗑️ Clear", variant="secondary", size="sm")
+                send_btn = gr.Button("🚀 Generate Tasks", variant="primary")
+                clear_btn = gr.Button("🗑️ Clear")
         
         with gr.Column(scale=1):
             gr.Markdown("### ⚙️ Generation Settings")
             
+            # 参数控制 - 简化配置
             max_tokens = gr.Slider(
                 minimum=50,
                 maximum=500,
                 value=200,
                 step=10,
-                label="Max Tokens",
-                info="Maximum number of tokens to generate"
+                label="Max Tokens"
             )
             
             temperature = gr.Slider(
@@ -207,8 +200,7 @@ with gr.Blocks(
                 maximum=2.0,
                 value=0.7,
                 step=0.1,
-                label="Temperature",
-                info="Controls randomness (lower = more focused)"
+                label="Temperature"
             )
             
             top_p = gr.Slider(
@@ -216,8 +208,7 @@ with gr.Blocks(
                 maximum=1.0,
                 value=0.9,
                 step=0.05,
-                label="Top-p",
-                info="Nucleus sampling threshold"
+                label="Top-p"
             )
             
             gr.Markdown("""
@@ -226,28 +217,29 @@ 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=['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.'],
+        examples=examples_data,
         inputs=msg,
         label="💡 Example Robot Commands"
     )
     
-    # 事件处理
+    # 事件绑定 - Gradio 3.x 风格
     msg.submit(
-        chat_interface,
+        fn=chat_interface,
         inputs=[msg, chatbot, max_tokens, temperature, top_p],
         outputs=[chatbot, msg]
     )
     
     send_btn.click(
-        chat_interface,
+        fn=chat_interface,
         inputs=[msg, chatbot, max_tokens, temperature, top_p],
         outputs=[chatbot, msg]
     )
     
     clear_btn.click(
-        lambda: ([], ""),
+        fn=lambda: ([], ""),
         outputs=[chatbot, msg]
     )
 

requirements.txt

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