Spaces:
Running
Running
# App | |
from get_predictions import get_predictions | |
import gradio | |
""" | |
@todo: refactor code + fix nltk download | |
@body : change file directory, | |
""" | |
if __name__ == "__main__": | |
interface = gradio.Interface( | |
get_predictions, | |
inputs=gradio.inputs.Textbox( | |
lines=2, | |
placeholder="Enter The Text", | |
default="", | |
label="Text to Predict", | |
), | |
outputs=[ | |
gradio.outputs.Textbox(type="auto", label="Aggression Prediction"), | |
gradio.outputs.Textbox(type="auto", label="Misogyny Prediction"), | |
], | |
title="Aggression and Misogyny Predictor", | |
theme="dark-huggingface", | |
live=True, | |
) | |
# Launch the interface | |
interface.launch( | |
share=False, | |
debug=True, | |
) | |