Update app.py

app.py

@@ -16,7 +16,9 @@ def ans(question ):
     gen = model.generate(inp, do_sample=True, top_p=0.9, temperature=0.86, max_new_tokens=100, repetition_penalty=1.2) #, stop_token="<eos>")
     
     gen = tokenizer.decode(gen[0])
-    return gen[:gen.index('<eos>') if '<eos>' in gen else len(gen)]
+    gen = gen[:gen.index('<eos>') if '<eos>' in gen else len(gen)]
+    gen = gem.split('Ответ:')[1]
+    return gen