update

app.py

@@ -222,11 +222,25 @@ def main():
         set_preset()
 
     st.title("GPT-JT")
-
-    col1, col2 = st.columns([1, 3])
-
+    
+    col1, col2 = st.columns([1, 2])
+    
     with col1:
         model_name = st.selectbox("Model", ["GPT-JT-6B-v1"])
+        
+    with col2:
+        preset = st.selectbox(
+            label="Examples", 
+            options=('Question Answering', 'Sentiment Analysis', 
+                     "Topic Classification", "Paraphrasing", "Text Summarization",
+                     "Word Sense Disambiguation", "Natural Language Inference"), 
+            on_change=set_preset,
+            key="preset",
+        )
+
+    col3, col4 = st.columns([1, 5])
+
+    with col3:
         max_new_tokens = st.text_input('Max new tokens', st.session_state.max_new_tokens)
         temperature = st.text_input('temperature', st.session_state.temperature)
         top_k = st.text_input('top_k', st.session_state.top_k)
@@ -237,16 +251,7 @@ def main():
         # seed = st.text_input('seed', "42")
         seed = "42"
 
-    with col2:
-
-        preset = st.selectbox(
-            label="Examples", 
-            options=('Question Answering', 'Sentiment Analysis', 
-                     "Topic Classification", "Paraphrasing", "Text Summarization",
-                     "Word Sense Disambiguation", "Natural Language Inference"), 
-            on_change=set_preset,
-            key="preset",
-        )
+    with col4:
 
         prompt_area = st.empty()
         prompt = prompt_area.text_area(