Update app.py

app.py

@@ -1,23 +1,22 @@
 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 = dspl
+    text = dataset.split()
     random.shuffle(text)
   
     for el, i in enumerate(text):