Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -19,7 +19,7 @@ KPI_FLAG_DEFAULT = os.getenv('KPI_FLAG', 'True') == 'True'
|
|
19 |
ENGAGEMENT_SCORE_DEFAULT = float(os.getenv('ENGAGEMENT_SCORE', '85.0'))
|
20 |
|
21 |
# Load model and tokenizer
|
22 |
-
model_name = "distilgpt2"
|
23 |
tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=True)
|
24 |
model = AutoModelForCausalLM.from_pretrained(model_name, low_cpu_mem_usage=True)
|
25 |
|
|
|
19 |
ENGAGEMENT_SCORE_DEFAULT = float(os.getenv('ENGAGEMENT_SCORE', '85.0'))
|
20 |
|
21 |
# Load model and tokenizer
|
22 |
+
model_name = "distilgpt2" # You can use a larger model here if needed
|
23 |
tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=True)
|
24 |
model = AutoModelForCausalLM.from_pretrained(model_name, low_cpu_mem_usage=True)
|
25 |
|