Update app.py

app.py

@@ -1,20 +1,3 @@
-import numpy as np
-import pandas as pd
-import matplotlib.pyplot as plt
-from matplotlib import style
-import seaborn as sns
-import tensorflow as tf
-from transformers import BertTokenizer
-
-
-
-from sklearn.model_selection import train_test_split
-from sklearn.preprocessing import LabelEncoder
-from keras.utils import to_categorical
-from sklearn import preprocessing
-import sklearn
-from sklearn.metrics import confusion_matrix
-
 from transformers import AutoModelForSequenceClassification
 from transformers import AutoTokenizer, AutoConfig
 import numpy as np
@@ -40,7 +23,7 @@ tokenizer = AutoTokenizer.from_pretrained(MODEL)
 config = AutoConfig.from_pretrained(MODEL)
 
 # create classifier function
-def classify_sentiments(text):
+def classify_compliant(text):
   text = preprocess(text)
   encoded_input = tokenizer(text, return_tensors='pt')
   output = model(**encoded_input)
@@ -69,7 +52,7 @@ article = """
             - Click clear button to refresh text
            """
 
-gr.Interface(,
+gr.Interface(classify_compliant,
             'text',
             'label',
             title = title,