Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
yuqiang
commited on
Commit
·
e4a24e9
1
Parent(s):
00373a8
fix index bug
Browse files- backend_api.py +1 -1
- config.py +1 -1
backend_api.py
CHANGED
@@ -9,7 +9,7 @@ from config import API_ENDPOINTS
|
|
9 |
def submit_to_backend(scene: str, episode: str, prompt: str, mode: str, model_type: str, user: str = "Gradio-user") -> dict:
|
10 |
job_id = str(uuid.uuid4())
|
11 |
|
12 |
-
scene_index = scene
|
13 |
episode_index = episode.split("_")[-1]
|
14 |
|
15 |
data = {
|
|
|
9 |
def submit_to_backend(scene: str, episode: str, prompt: str, mode: str, model_type: str, user: str = "Gradio-user") -> dict:
|
10 |
job_id = str(uuid.uuid4())
|
11 |
|
12 |
+
scene_index = scene[-1]
|
13 |
episode_index = episode.split("_")[-1]
|
14 |
|
15 |
data = {
|
config.py
CHANGED
@@ -68,5 +68,5 @@ EPISODE_CONFIGS = {
|
|
68 |
}
|
69 |
}
|
70 |
|
71 |
-
MODEL_CHOICES = ["
|
72 |
MODE_CHOICES = ["vlnPE", "vlnCE"]
|
|
|
68 |
}
|
69 |
}
|
70 |
|
71 |
+
MODEL_CHOICES = ["InternVLA-N1", "rdp", "cma"]
|
72 |
MODE_CHOICES = ["vlnPE", "vlnCE"]
|