fix heights

app.py

@@ -143,9 +143,9 @@ def display_llm_output():
     form = st.form('llm')
 
     prompt_format_str = get_prompt_format(st.session_state["model_name_or_path"])
-    prompt_format = form.text_area('Prompt format', value=prompt_format_str, height=200)
-    system_prompt = form.text_area('System message', value=st.session_state["system_prompt"], height=200)
-    prompt = form.text_area('Prompt', value=st.session_state["prompt"], height=200)
+    prompt_format = form.text_area('Prompt format', value=prompt_format_str, height=170)
+    system_prompt = form.text_area('System message', value=st.session_state["system_prompt"], height=170)
+    prompt = form.text_area('Prompt', value=st.session_state["prompt"], height=170)
 
     submitted = form.form_submit_button('Submit')
 

utils/epfl_meditron_utils.py

@@ -47,7 +47,8 @@ def get_llm_response(model_name_or_path, temperature, do_sample, top_p, top_k, m
         temperature=temperature,
         top_p=top_p,
         top_k=top_k,
-        repetition_penalty=repetition_penalty
+        repetition_penalty=repetition_penalty,
+        return_full_text=False
     )
 
     pipe_response = pipe(formatted_prompt)