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} | |
TITLE = "AutoCounterspeech" | |
DESCRIPTION = "Enter some hate speech, and see if the system generates an appropriate counterspeech response." | |
ARTICLE = f""" | |
**Model:** {DEFAULT_MODEL}<br> | |
**Decoding parameters:** {DECODING_CONFIG} | |
""" | |
model = ResponseGenerator(DEFAULT_MODEL, DECODING_CONFIG) | |
def respond(input): | |
return model.respond(input) | |
demo = gr.Interface(fn=respond, inputs="text", outputs="text", title = TITLE, description = DESCRIPTION, article = ARTICLE) | |
demo.launch() | |