fix version bug

app.py

@@ -11,7 +11,10 @@ max_index = 108
 if 'index' not in st.session_state:
     st.session_state.index = 1
 
-buttons = st.columns([2, 1, 6])
+buttons = st.columns([2, 1.1, 5.9])
+
+with buttons[0]:
+    st.markdown("# HumanEval-V Viewer")
 
 with buttons[1]:
     # Number input for navigation
@@ -22,20 +25,10 @@ with buttons[1]:
         value=st.session_state.index, 
         key="index_input", 
         help="Enter an index and jump to that index.",
-        step=1  # Increment by 1
+        step=1
     )
 
-with buttons[0]:
-    st.markdown("# HumanEval-V Viewer")
-    
-
-# Check if the input differs from the current session state and update it
-if index_input != st.session_state.index:
-    st.session_state.index = index_input
-    st.experimental_rerun()
-
-
-coding_task = humaneval_v_data[st.session_state.index-1]
+coding_task = humaneval_v_data[index_input-1]
 qid = coding_task["qid"]
 image = coding_task["image"]
 function_signature = coding_task["function_signature"]