Updated app.py.

app.py

@@ -26,15 +26,15 @@ def process_text(text):
 vectorizer = CountVectorizer()
 def vectorize_text(text):
     text = process_text(text)
-    text = [text]
     return vectorizer.transform(text)
 
 # Load model from pickle file
 model = pickle.load(open(lr_filename, 'rb'))
 
-
 def predict(text):
-    return model.predict([text])
+    text = vectorize_text(text)
+    prediction = model.predict(text)
+    return prediction
 
 
 # Define interface
@@ -48,5 +48,3 @@ demo = gr.Interface(fn=predict,
 
 demo.launch()
 
-
-