Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 =
|
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']
|