Delete ipynb

app.py

@@ -22,8 +22,8 @@ def load_model_tokenizer():
 
 
 @st.cache
-def load_pipeline(checkpoint):
-    return pipeline("question-answering", model=checkpoint)
+def load_pipeline(model_cp, tokenizer_cp):
+    return pipeline("question-answering", model=model_cp, tokenizer=tokenizer_cp)
 
 
 # Page config
@@ -39,17 +39,12 @@ if uploaded_file is not None:
     st.write(df.head())
 
 # Load tokenizer and model
-checkpoint = "aidan-o-brien/recipe-improver"
-question_answer = load_pipeline(checkpoint)
+model_cp = "aidan-o-brien/recipe-improver"
+tokenizer_cp = "albert-base-v2"
+question_answer = load_pipeline(model_cp, tokenizer_cp)
 st.write("Model and tokenizer successfully loaded.")
 
-# Pre-process data from csv file
-
-
 # Run inference
-
-
-# Post-process output of model
-
+first_example = df['review'][0]
 
 # Present results