Update app.py

app.py

@@ -64,7 +64,7 @@ bertscore = load_metric('bertscore')
 MAX_INPUT_LENGTH = 256
 MAX_TARGET_LENGTH = 128
 
-def preprocess_function(examples):    
+def preprocess_function(examples, tokenizer):    
     """
     Preprocess entries of the given dataset
 
@@ -147,7 +147,7 @@ def extract_labels(predictions):
     return labels
 
 
-def get_predictions_labels(model, dataloader):
+def get_predictions_labels(model, dataloader, tokenizer):
     """
     Evaluate model on the given dataset
 
@@ -207,7 +207,7 @@ def load_data():
 
         dataloader = DataLoader(processed_dataset, batch_size=4)
 
-        predictions, labels = get_predictions_labels(model, dataloader)
+        predictions, labels = get_predictions_labels(model, dataloader, tokenizer)
 
         predicted_feedback = extract_feedback(predictions)
         predicted_labels = extract_labels(predictions)