Spaces:

AlexWortega
/

MailruQA

Running

AlexWortega commited on Jul 7, 2022

Commit

9421b11

•

1 Parent(s): a9751ec

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ def ans(question ):
     gen = tokenizer.decode(gen[0])
     gen = gen[:gen.index('<eos>') if '<eos>' in gen else len(gen)]
-    gen = gem.split('Ответ:')[1]
     return gen

     gen = tokenizer.decode(gen[0])
     gen = gen[:gen.index('<eos>') if '<eos>' in gen else len(gen)]
+    gen = gen.split('Ответ:')[1]
     return gen