Spaces:
Runtime error
Runtime error
Shane Weisz
commited on
Commit
•
6f57bdb
1
Parent(s):
46cb133
Switch to 5-gram blocking
Browse files
app.py
CHANGED
@@ -2,7 +2,7 @@ from response_generation import ResponseGenerator
|
|
2 |
import gradio as gr
|
3 |
|
4 |
DEFAULT_MODEL = "shaneweisz/DialoGPT-finetuned-multiCONAN"
|
5 |
-
DECODING_CONFIG = {"max_new_tokens": 100, "min_new_tokens": 20, "no_repeat_ngram_size":
|
6 |
|
7 |
TITLE = "Automating Counterspeech in Dialogue Systems"
|
8 |
DESCRIPTION = """
|
|
|
2 |
import gradio as gr
|
3 |
|
4 |
DEFAULT_MODEL = "shaneweisz/DialoGPT-finetuned-multiCONAN"
|
5 |
+
DECODING_CONFIG = {"max_new_tokens": 100, "min_new_tokens": 20, "no_repeat_ngram_size": 5, "num_beams": 10}
|
6 |
|
7 |
TITLE = "Automating Counterspeech in Dialogue Systems"
|
8 |
DESCRIPTION = """
|