Update app.py

app.py

@@ -25,9 +25,11 @@ def generate(tokenizer, model, text, features):
        
         if '<|authornameend|>' not in raw:
             continue
-      
-        author, text = raw.split('<|authornameend|>')
-     
+        
+        end_name = raw.split('<|authornameend|>')      
+        author = end_name[-2]
+        text = end_name[-1]
+    
         count += 1
         st.markdown('**' + author.strip() + '**: ' + text.replace('<|endoftext|>', '').replace('<|pad|>', '').strip())