Update app.py

app.py

@@ -1,5 +1,6 @@
 from transformers import AutoModelForSequenceClassification
 from transformers import AutoTokenizer, AutoConfig
+from clean_data import cleaned_complaints
 import numpy as np
 from scipy.special import softmax
 import gradio as gr
@@ -24,7 +25,8 @@ config = AutoConfig.from_pretrained(MODEL)
 
 # create classifier function
 def classify_compliant(text):
-  text = preprocess(text)
+  text_clean = cleaned_complaints(text)
+  text = preprocess(text_clean)
   encoded_input = tokenizer(text, return_tensors='pt')
   output = model(**encoded_input)
   scores = output[0][0].detach().numpy()