Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,7 @@ import transformers
|
|
4 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
5 |
|
6 |
SAVED_CHECKPOINT = 'mikegarts/distilgpt2-erichmariaremarque'
|
7 |
-
MIN_WORDS =
|
8 |
|
9 |
|
10 |
def get_model():
|
@@ -47,5 +47,5 @@ gr.Interface(
|
|
47 |
outputs="text",
|
48 |
title=title,
|
49 |
description=description,
|
50 |
-
examples=[["I was drinking because"], ["Who is Karl for me?"]]
|
51 |
).launch(debug=True)
|
|
|
4 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
5 |
|
6 |
SAVED_CHECKPOINT = 'mikegarts/distilgpt2-erichmariaremarque'
|
7 |
+
MIN_WORDS = 100
|
8 |
|
9 |
|
10 |
def get_model():
|
|
|
47 |
outputs="text",
|
48 |
title=title,
|
49 |
description=description,
|
50 |
+
examples=[["I was drinking because"], ["Who is Karl for me?"], ["My most brutal mistake was"]]
|
51 |
).launch(debug=True)
|