Update app.py

app.py

@@ -4,12 +4,14 @@ import random
 
 endings = ['ый', 'ий', 'ая', 'яя', 'ое', 'ее', 'ые', 'ие', 'ому', 'ему', 'ой', 'ей', 'ых', 'их', 'ым', 'им', 'ого', 'его', 'ую', 'юю', 'ого', 'его']
 
+with open("dataset.txt", "r") as f:
+    _dataset = f.readlines()
+__dataset = {el: {"content": i} for el, i in enumerate(_dataset)}
+
 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: