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']
|