Add prediction functionality

app.py

@@ -8,8 +8,9 @@ os.system("cp -r ./nltk_data/ /home/user/nltk_data")
 def analyze(text):
     model = CustomModel()
     text = preprocess_pipeline(text)
-    text = vectorizer([text])
-    return text.shape
+    vector = vectorizer([text])
+    pred = model.predict(vector)
+    return pred
 
 app = gr.Interface(fn=analyze, inputs="text", outputs="text")