Pinpoint-Web / app.py
User1342's picture
Update app.py
48a493e
import nltk
nltk.download('averaged_perceptron_tagger')
import gradio as gr
import predictor
def check_string(string_to_predict):
try:
is_extremist = predictor.predictor().predict(string_to_predict)
if is_extremist:
return "The message has been identified as potentially containing violent far-right extremist content."
else:
return "The message has been identified as not containing violent far-right extremist content."
except FileNotFoundError as e:
return "The message was not feature rich enough to identify, try something else."
demo = gr.Interface(
fn=check_string,
inputs=gr.Textbox(lines=2, placeholder="Text to predict here..."),
outputs="text",
)
demo.launch()