Update app.py

app.py

@@ -38,6 +38,8 @@ def plot_word_scores(top_token_ids, outputs, tokenizer, boolq=False, width=600):
 
 st.title('How do LLM choose their words?')
 
+instruction = st.text_area(label='Write an instruction:', placeholder='Where is Venice located?')
+
 col1, col2 = st.columns(2)
 
 with col1:
@@ -62,8 +64,6 @@ model = T5ForConditionalGeneration.from_pretrained(
     device_map="auto"
 )
 
-instruction = st.text_area('Write an instruction:', 'Where is Venice located?')
-
 
 prompts = [
     f"""Below is an instruction that describes a task. Write a response that appropriately completes the request.