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
|