yuqiang commited on
Commit
e4a24e9
·
1 Parent(s): 00373a8

fix index bug

Browse files
Files changed (2) hide show
  1. backend_api.py +1 -1
  2. 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.split("_")[-1]
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 = ["rdp", "cma", "InternVLA-N1"]
72
  MODE_CHOICES = ["vlnPE", "vlnCE"]
 
68
  }
69
  }
70
 
71
+ MODEL_CHOICES = ["InternVLA-N1", "rdp", "cma"]
72
  MODE_CHOICES = ["vlnPE", "vlnCE"]