Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -6,7 +6,7 @@ from transformers import pipeline
|
|
6 |
generator = pipeline('text-generation', model='distilgpt2') # Lightweight model
|
7 |
|
8 |
# Predefined words to check
|
9 |
-
SPECIAL_WORDS = ['weather', 'sun', 'middle', '
|
10 |
|
11 |
# Global variable to store the initial design
|
12 |
initial_word_design = ""
|
|
|
6 |
generator = pipeline('text-generation', model='distilgpt2') # Lightweight model
|
7 |
|
8 |
# Predefined words to check
|
9 |
+
SPECIAL_WORDS = ['weather', 'sun', 'middle', 'spring', 'heat']
|
10 |
|
11 |
# Global variable to store the initial design
|
12 |
initial_word_design = ""
|