srivarshan's picture
Add prediction functionality
dc40079
raw
history blame
437 Bytes
import gradio as gr
from model import CustomModel
from preprocess import preprocess_pipeline, vectorizer
import os
os.system("cp -r ./nltk_data/ /home/user/nltk_data")
def analyze(text):
model = CustomModel()
text = preprocess_pipeline(text)
vector = vectorizer([text])
pred = model.predict(vector)
return pred
app = gr.Interface(fn=analyze, inputs="text", outputs="text")
# app.launch(share="True")
app.launch()