Change order and run inference on one example

app.py

@@ -23,11 +23,6 @@ icon = "🍣"
 st.set_page_config(page_title=title, page_icon=icon)
 st.title(title)
 
-# Load csv
-uploaded_file = st.file_uploader("Choose a csv file", type="csv", key='file_uploader')
-if uploaded_file is not None:
-    df = load_data(uploaded_file)
-    st.write(df.head())
 
 # Load tokenizer and model
 model_cp = "aidan-o-brien/recipe-improver"
@@ -35,7 +30,17 @@ tokenizer_cp = "albert-base-v2"
 question_answer = load_pipeline(model_cp, tokenizer_cp)
 st.write("Model and tokenizer successfully loaded.")
 
-# Run inference
-first_example = df['review'][0]
 
-# Present results
+# Load csv
+uploaded_file = st.file_uploader("Choose a csv file", type="csv", key='file_uploader')
+if uploaded_file is not None:
+    df = load_data(uploaded_file)
+    st.write(df.head())
+
+    # Run inference on first example
+    first_example = df['review'][0]
+    question = "how to improve this recipe?"
+    answer = question_answer(question=question, context=first_example)
+
+    # Present results
+    st.write(answer)