Update app.py

app.py

@@ -44,26 +44,30 @@ def infer(prompt, model_name, max_new_tokens=10, temperature=0.0, top_p=1.0):
     
     
 st.title("TOMA Application")
- 
-s_example = "Please answer the following question:\n\nQuestion: Where is Zurich?\nAnswer:"
-prompt = st.text_area(
-    "Prompt",
-    value=s_example,
-    max_chars=4096,
-    height=400,
-)
-    
-generated_area = st.empty()
-generated_area.markdown("(Generate here)")
 
-button_submit = st.button("Submit")
-   
-model_name = st.selectbox("Model", ["GPT-JT-6B-v1"])
-max_new_tokens = st.text_input('Max new tokens', "10")
-temperature = st.text_input('temperature', "0.0")
-top_p = st.text_input('top_p', "1.0")
+col1, col2 = st.columns([1, 3])
 
-if button_submit:
+with col2:
+    s_example = "Please answer the following question:\n\nQuestion: Where is Zurich?\nAnswer:"
+    prompt = st.text_area(
+        "Prompt",
+        value=s_example,
+        max_chars=4096,
+        height=400,
+    )
+        
+    generated_area = st.empty()
+    generated_area.markdown("(Generate here)")
+    
+    button_submit = st.button("Submit")
+
+    if button_submit:
     with st.spinner(text="In progress.."):
         report_text = infer(prompt, model_name=model_name, max_new_tokens=max_new_tokens, temperature=temperature, top_p=top_p)
-        generated_area.markdown(report_text)
+        generated_area.markdown(report_text)
+
+with col1:
+    model_name = st.selectbox("Model", ["GPT-JT-6B-v1"])
+    max_new_tokens = st.text_input('Max new tokens', "10")
+    temperature = st.text_input('temperature', "0.0")
+    top_p = st.text_input('top_p', "1.0")