Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -7,13 +7,13 @@ from huggingface_hub import upload_file, InferenceClient
|
|
| 7 |
from PIL import Image
|
| 8 |
|
| 9 |
# --- Config ---
|
| 10 |
-
HF_TOKEN = os.environ.get("HF_CV_ROBOT_TOKEN")
|
| 11 |
HF_DATASET_REPO = "OppaAI/Robot_MCP"
|
| 12 |
# Model specifically for VLM (image-to-text) tasks on Hugging Face
|
| 13 |
HF_VLM_MODEL = "Qwen/Qwen2.5-VL-7B-Instruct"
|
| 14 |
|
| 15 |
-
if not HF_TOKEN:
|
| 16 |
-
raise ValueError("HF_TOKEN environment variable not set.")
|
| 17 |
|
| 18 |
# Initialize the Hugging Face Inference Client
|
| 19 |
hf_client = InferenceClient(token=HF_TOKEN)
|
|
@@ -42,6 +42,7 @@ def save_and_upload_image(image_b64):
|
|
| 42 |
# --- Main MCP function ---
|
| 43 |
def process_and_describe(payload: dict):
|
| 44 |
try:
|
|
|
|
| 45 |
robot_id = payload.get("robot_id", "unknown")
|
| 46 |
image_b64 = payload["image_b64"]
|
| 47 |
|
|
|
|
| 7 |
from PIL import Image
|
| 8 |
|
| 9 |
# --- Config ---
|
| 10 |
+
#HF_TOKEN = os.environ.get("HF_CV_ROBOT_TOKEN")
|
| 11 |
HF_DATASET_REPO = "OppaAI/Robot_MCP"
|
| 12 |
# Model specifically for VLM (image-to-text) tasks on Hugging Face
|
| 13 |
HF_VLM_MODEL = "Qwen/Qwen2.5-VL-7B-Instruct"
|
| 14 |
|
| 15 |
+
#if not HF_TOKEN:
|
| 16 |
+
# raise ValueError("HF_TOKEN environment variable not set.")
|
| 17 |
|
| 18 |
# Initialize the Hugging Face Inference Client
|
| 19 |
hf_client = InferenceClient(token=HF_TOKEN)
|
|
|
|
| 42 |
# --- Main MCP function ---
|
| 43 |
def process_and_describe(payload: dict):
|
| 44 |
try:
|
| 45 |
+
hf_token = payload.get("hf_token", HF_TOKEN)
|
| 46 |
robot_id = payload.get("robot_id", "unknown")
|
| 47 |
image_b64 = payload["image_b64"]
|
| 48 |
|