Update app.py

app.py

@@ -22,7 +22,8 @@ def load_v2t_samples(data_root):
 
 def load_t2v_samples(data_root):
     sample_text = ['cut the sausage', 'stir vegetables into salmon', 'rinse cutting board']
-    idx2sid = {0: 2119, 1: 1730, 2: 1276}
+    #idx2sid = {0: 2119, 1: 1730, 2: 1276}
+    idx2sid = {0: 2119, 2: 1276}
     return sample_text, idx2sid
 
 def format_pred(pred, gt):
@@ -45,6 +46,8 @@ def main():
     print(v2t_samples)
 
     def predict_v2t(idx):
+        if idx == 1:
+            idx = 2
         sid = idx2sid_v2t[idx]
         zeroshot_action, gt_action = lavila.predict_v2t(idx, sid)
         egovpa_action, gt_action = egovpa.predict_v2t(idx, sid)