change model loading to pipeline function

app.py

@@ -1,9 +1,9 @@
 from distutils.command.upload import upload
-from io import StringIO
 import pandas as pd
 import streamlit as st
 
 from transformers import AutoTokenizer, TFAutoModelForQuestionAnswering
+from transformers import pipeline
 
 
 @st.cache
@@ -21,6 +21,11 @@ def load_model_tokenizer():
     return model, tokenizer
 
 
+@st.cache
+def load_pipeline(checkpoint):
+    return pipeline("question-answering", model=checkpoint)
+
+
 # Page config
 title = "Recipe Improver"
 icon = "🍣"
@@ -34,7 +39,8 @@ if uploaded_file is not None:
     st.write(df.head())
 
 # Load tokenizer and model
-model, tokenizer = load_model_tokenizer()
+checkpoint = "aidan-o-brien/recipe-improver"
+question_answer = load_pipeline(checkpoint)
 st.write("Model and tokenizer successfully loaded.")
 
 # Pre-process data from csv file