pip64 ierhon commited on
Commit
7489548
·
1 Parent(s): 67894b9

Update app.py (#2)

Browse files

- Update app.py (d47a40936d6b2bbf04161d782769d5e690380d25)


Co-authored-by: - - - <[email protected]>

Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -5,11 +5,14 @@ import random
5
  endings = ['ый', 'ий', 'ая', 'яя', 'ое', 'ее', 'ые', 'ие', 'ому', 'ему', 'ой', 'ей', 'ых', 'их', 'ым', 'им', 'ого', 'его', 'ую', 'юю', 'ого', 'его']
6
  _dataset = open("dataset.txt", "r").readlines()
7
 
 
 
 
 
8
  def gen(text):
9
  a = text
10
  count = random.randint(5, 15)
11
  _a = list(set(a))
12
- __dataset = {el: {"content": i} for el, i in enumerate(_dataset)}
13
  dataset = ' '.join([i['content'].replace('\n', '') for i in __dataset.values() if any(ii in i['content'] for ii in _a)])
14
 
15
  if len(dataset.strip()) == 0:
 
5
  endings = ['ый', 'ий', 'ая', 'яя', 'ое', 'ее', 'ые', 'ие', 'ому', 'ему', 'ой', 'ей', 'ых', 'их', 'ым', 'им', 'ого', 'его', 'ую', 'юю', 'ого', 'его']
6
  _dataset = open("dataset.txt", "r").readlines()
7
 
8
+ with open("dataset.txt", "r") as f:
9
+ _dataset = f.readlines()
10
+ __dataset = {el: {"content": i} for el, i in enumerate(_dataset)}
11
+
12
  def gen(text):
13
  a = text
14
  count = random.randint(5, 15)
15
  _a = list(set(a))
 
16
  dataset = ' '.join([i['content'].replace('\n', '') for i in __dataset.values() if any(ii in i['content'] for ii in _a)])
17
 
18
  if len(dataset.strip()) == 0: