Spaces:
Runtime error
Runtime error
| 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() | |