Update app.py

app.py

@@ -1,22 +1,23 @@
 import gradio as gr
-
 import random
 
 endings = ['ый', 'ий', 'ая', 'яя', 'ое', 'ее', 'ые', 'ие', 'ому', 'ему', 'ой', 'ей', 'ых', 'их', 'ым', 'им', 'ого', 'его', 'ую', 'юю', 'ого', 'его']
 
+_dataset = open("dataset.txt", "r").readlines()
+_a = list(set(a))
+__dataset = {el: {"content": i} for el, i in enumerate(_dataset)}
+dataset = ' '.join([i['content'].replace('\n', '') for i in __dataset.values() if any(ii in i['content'] for ii in _a)])
+dspl = dataset.split()
+
 def gen(text):
     a = text
     count = random.randint(5, 15)
-    _dataset = open("dataset.txt", "r").readlines()
-    _a = list(set(a))
-    __dataset = {el: {"content": i} for el, i in enumerate(_dataset)}
-    dataset = ' '.join([i['content'].replace('\n', '') for i in __dataset.values() if any(ii in i['content'] for ii in _a)])
     
     if len(dataset.strip()) == 0:
         return random.choice(_dataset).strip()
   
     words = {}
-    text = dataset.split()
+    text = dspl
     random.shuffle(text)
   
     for el, i in enumerate(text):