doevent commited on
Commit
c427172
1 Parent(s): 8d74675

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,8 +9,8 @@ def generate(starting_text):
9
  seed = random.randint(1, 100000)
10
  set_seed(seed)
11
  while True:
12
- response = str(gpt2_pipe(starting_text, max_length=30, num_return_sequences=random.randint(5, 15))).strip()
13
- if starting_text != response[1]['generated_text']:
14
  print(f"Repeat: {response}")
15
  else:
16
  return response[1]['generated_text']
 
9
  seed = random.randint(1, 100000)
10
  set_seed(seed)
11
  while True:
12
+ response = gpt2_pipe(starting_text, max_length=30, num_return_sequences=random.randint(5, 15))
13
+ if starting_text != response[1]['generated_text'].strip():
14
  print(f"Repeat: {response}")
15
  else:
16
  return response[1]['generated_text']