Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -3,8 +3,8 @@ import random
|
|
3 |
from transformers import pipeline
|
4 |
|
5 |
# Load the GPT2 model
|
6 |
-
|
7 |
-
generator = pipeline('text-generation', model='bloom-560m')
|
8 |
|
9 |
# Set the maximum length of the generated prompt
|
10 |
max_length = 50
|
|
|
3 |
from transformers import pipeline
|
4 |
|
5 |
# Load the GPT2 model
|
6 |
+
generator = pipeline('text-generation', model='gpt2')
|
7 |
+
#generator = pipeline('text-generation', model='bloom-560m')
|
8 |
|
9 |
# Set the maximum length of the generated prompt
|
10 |
max_length = 50
|