Update app.py

app.py

@@ -15,7 +15,9 @@ def generate(tokenizer, model, text, features):
     )
     for i, sample_output in enumerate(sample_outputs):
         decoded = tokenizer.decode(sample_output, skip_special_tokens=False)
-        autor, text = decoded.split('<|authornamebegin|>')[1].split('<|authornameend|>')
+        if '<|authornameend|>' not in decoded:
+            continue
+        author, text = decoded.split('<|authornamebegin|>')[1].split('<|authornameend|>')
         st.markdown('**' + author.strip() + '**: ' + text.replace('<|endoftext|>', '').replace('<|pad|>', '').strip())