Updated app.py.

app.py

@@ -6,9 +6,9 @@ import pickle5 as pickle
 # Load model from pickle file
 model = pickle.load(open('lg_classifier.sav', 'rb'))
 
-# Define function to predict
+# Define function to predict label
 def predict(text):
-    return model.predict([text])
+    return model.predict([[text]])[0]
 
 # Define interface
 demo = gr.Interface(fn=predict,