from response_generation import ResponseGenerator
import gradio as gr


DEFAULT_MODEL = "shaneweisz/DialoGPT-finetuned-multiCONAN"
DECODING_CONFIG = {"max_new_tokens": 100, "no_repeat_ngram_size": 3, "num_beams": 10}

model = ResponseGenerator(DEFAULT_MODEL, DECODING_CONFIG)


def respond(hate_speech_input_text):
    return model.respond(hate_speech_input_text)


demo = gr.Interface(fn=respond, inputs="text", outputs="text")

demo.launch()