OppaAI commited on
Commit
51deb36
·
verified ·
1 Parent(s): 49eb2ad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
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