Spaces:
Sleeping
Sleeping
yongdong
commited on
Commit
·
6158c62
1
Parent(s):
1f6f70b
Update lora model name.
Browse files
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/
|
| 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/
|
| 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=[
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 243 |
inputs=msg,
|
| 244 |
-
label="💡 Example
|
| 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
|