Updated app.py.

app.py

@@ -3,12 +3,16 @@ import gradio as gr
 from sklearn.linear_model import LogisticRegression
 import pickle5 as pickle
 
+# file name
+lr_filename = 'lg_classifier.sav'
+
 # Load model from pickle file
-model = pickle.load(open('lg_classifier.sav', 'rb'))
+model = pickle.load(open(lr_filename, 'rb'))
 
-# Define function to predict label
+# Define function to make a prediction with the model
 def predict(text):
-    return model.predict([[text]])[0]
+    return model.predict([text])[0]
+
 
 # Define interface
 demo = gr.Interface(fn=predict,