doevent commited on
Commit
68b77e7
β€’
1 Parent(s): 926e92f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -15,14 +15,15 @@ def generate(starting_text):
15
 
16
  # If the text field is empty
17
  if starting_text == "":
18
- starting_text: str = line[random.randrange(0, len(line))].replace("\n", "")
19
  starting_text: str = re.sub(r"[,:\-–.!;?_]", '', starting_text)
20
  print(starting_text)
21
 
22
  response = gpt2_pipe(starting_text, max_length=random.randint(20, 45), num_return_sequences=random.randint(5, 15))
23
  response_list = []
24
  for x in response:
25
- if x['generated_text'].strip() != starting_text and len(x['generated_text'].strip()) > (len(starting_text) + 4):
 
26
  response_list.append(x['generated_text'])
27
 
28
  response_end = "\n".join(response_list)
 
15
 
16
  # If the text field is empty
17
  if starting_text == "":
18
+ starting_text: str = line[random.randrange(0, len(line))].replace("\n", "").lower().capitalize()
19
  starting_text: str = re.sub(r"[,:\-–.!;?_]", '', starting_text)
20
  print(starting_text)
21
 
22
  response = gpt2_pipe(starting_text, max_length=random.randint(20, 45), num_return_sequences=random.randint(5, 15))
23
  response_list = []
24
  for x in response:
25
+ if x['generated_text'].strip() != starting_text and len(x['generated_text'].strip()) > (len(starting_text) + 4) \
26
+ and tr.endswith((":", "-", "β€”")) is False:
27
  response_list.append(x['generated_text'])
28
 
29
  response_end = "\n".join(response_list)