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)
|