MesonWarrior commited on
Commit
c20a298
1 Parent(s): 4ffb6ee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -1
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
- return pipe(text)[0]['generated_text']
 
 
 
 
 
 
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():