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"]
|