yongdong commited on
Commit
6158c62
·
1 Parent(s): 1f6f70b

Update lora model name.

Browse files
Files changed (1) hide show
  1. app.py +12 -4
app.py CHANGED
@@ -19,7 +19,7 @@ warnings.filterwarnings("ignore")
19
 
20
  # Model configuration
21
  MODEL_NAME = "meta-llama/Llama-3.1-8B"
22
- LORA_MODEL = "YongdongWang/llama-3.1-8b-dart-qlora"
23
 
24
  # Global variables to store model and tokenizer
25
  model = None
@@ -189,7 +189,7 @@ with gr.Blocks(
189
 
190
  **Capabilities**: Convert natural language robot commands into structured task sequences for excavators, dump trucks, and other construction robots.
191
 
192
- **Model**: [YongdongWang/llama-3.1-8b-dart-qlora](https://huggingface.co/YongdongWang/llama-3.1-8b-dart-qlora)
193
 
194
  ⚡ **Using ZeroGPU**: This Space uses dynamic GPU allocation (Nvidia H200). First generation might take a bit longer.
195
  """)
@@ -239,9 +239,17 @@ with gr.Blocks(
239
 
240
  # Example conversations
241
  gr.Examples(
242
- 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.'],
 
 
 
 
 
 
 
 
243
  inputs=msg,
244
- label="💡 Example Robot Commands"
245
  )
246
 
247
  # Event handling
 
19
 
20
  # Model configuration
21
  MODEL_NAME = "meta-llama/Llama-3.1-8B"
22
+ LORA_MODEL = "YongdongWang/llama3.1-8b-lora-qlora-dart-llm"
23
 
24
  # Global variables to store model and tokenizer
25
  model = None
 
189
 
190
  **Capabilities**: Convert natural language robot commands into structured task sequences for excavators, dump trucks, and other construction robots.
191
 
192
+ **Model**: [YongdongWang/llama3.1-8b-lora-qlora-dart-llm](https://huggingface.co/YongdongWang/llama3.1-8b-lora-qlora-dart-llm)
193
 
194
  ⚡ **Using ZeroGPU**: This Space uses dynamic GPU allocation (Nvidia H200). First generation might take a bit longer.
195
  """)
 
239
 
240
  # Example conversations
241
  gr.Examples(
242
+ examples=[
243
+ "Deploy Excavator 1 to Soil Area 1 for excavation.",
244
+ "Send Dump Truck 1 to collect material from Excavator 1, then unload at the designated area.",
245
+ "Move all robots to avoid Puddle 1 after inspection is complete.",
246
+ "Deploy Excavator 2 with Dump Truck 2 to Soil Area 3, then return both to start position.",
247
+ "Send Robot 1 to inspect Rock Area 1, then avoid that area with all other robots.",
248
+ "Execute excavation at multiple soil areas using different excavators simultaneously.",
249
+ "Coordinate Dump Truck 1 and Dump Truck 2 to transport materials from Excavation Site 1 to Storage Area 1.",
250
+ ],
251
  inputs=msg,
252
+ label="💡 Example Operator Commands"
253
  )
254
 
255
  # Event handling