Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -6,7 +6,7 @@ import json
|
|
| 6 |
|
| 7 |
# --- Config ---
|
| 8 |
HF_DATASET_REPO = "OppaAI/Robot_MCP"
|
| 9 |
-
HF_VLM_MODEL = "Qwen/Qwen2.5-VL-
|
| 10 |
|
| 11 |
# --- Helper Functions ---
|
| 12 |
def save_and_upload_image(image_b64, hf_token):
|
|
@@ -50,9 +50,9 @@ def process_and_describe(payload: dict):
|
|
| 50 |
# System prompt: describe + suggest action
|
| 51 |
system_prompt = """
|
| 52 |
You are a helpful robot assistant.
|
| 53 |
-
1. Describe the image in detail.
|
| 54 |
2. Suggest what the robot should do next based on what it sees:
|
| 55 |
-
- Human figure β
|
| 56 |
- Ball β move towards it.
|
| 57 |
- Obstacles β stop or avoid.
|
| 58 |
- Animal β identify the animal and take photos
|
|
|
|
| 6 |
|
| 7 |
# --- Config ---
|
| 8 |
HF_DATASET_REPO = "OppaAI/Robot_MCP"
|
| 9 |
+
HF_VLM_MODEL = "Qwen/Qwen2.5-VL-7B-Instruct"
|
| 10 |
|
| 11 |
# --- Helper Functions ---
|
| 12 |
def save_and_upload_image(image_b64, hf_token):
|
|
|
|
| 50 |
# System prompt: describe + suggest action
|
| 51 |
system_prompt = """
|
| 52 |
You are a helpful robot assistant.
|
| 53 |
+
1. Describe the image in detail, especially humans in full detail.
|
| 54 |
2. Suggest what the robot should do next based on what it sees:
|
| 55 |
+
- Human figure β say 'Hi'.
|
| 56 |
- Ball β move towards it.
|
| 57 |
- Obstacles β stop or avoid.
|
| 58 |
- Animal β identify the animal and take photos
|