Updated app.py.

app.py

@@ -8,7 +8,7 @@ model = pickle.load(open('lg_classifier.sav', 'rb'))
 
 # Define function to predict
 def predict(text):
-    return model.predict([[text]])[0]
+    return model.predict([text])
 
 # Define interface
 demo = gr.Interface(fn=predict,