Spaces:
Sleeping
Sleeping
MesonWarrior
commited on
Commit
•
c20a298
1
Parent(s):
4ffb6ee
Update app.py
Browse files
app.py
CHANGED
@@ -17,10 +17,17 @@ def generate_text(
|
|
17 |
tokenizer='MesonWarrior/gpt2-vk-bugro',
|
18 |
min_length=min_length,
|
19 |
max_length=max_length,
|
|
|
20 |
use_auth_token=True
|
21 |
)
|
22 |
|
23 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
24 |
|
25 |
def interface():
|
26 |
with gr.Row():
|
|
|
17 |
tokenizer='MesonWarrior/gpt2-vk-bugro',
|
18 |
min_length=min_length,
|
19 |
max_length=max_length,
|
20 |
+
do_not_truncate=do_not_truncate,
|
21 |
use_auth_token=True
|
22 |
)
|
23 |
|
24 |
+
print('generating...')
|
25 |
+
|
26 |
+
output = pipe(text)
|
27 |
+
|
28 |
+
print(output)
|
29 |
+
|
30 |
+
return output[0]['generated_text']
|
31 |
|
32 |
def interface():
|
33 |
with gr.Row():
|