Update app.py

app.py

@@ -18,8 +18,8 @@ if missing_vars:
 KPI_FLAG_DEFAULT = os.getenv('KPI_FLAG', 'True') == 'True'
 ENGAGEMENT_SCORE_DEFAULT = float(os.getenv('ENGAGEMENT_SCORE', '85.0'))
 
-# Load model and tokenizer
-model_name = "gpt2-medium"  # Using gpt2-medium as previously updated
+# Load model and tokenizer (Updated to use distilgpt2)
+model_name = "distilgpt2"  # Using distilgpt2
 tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=True)
 model = AutoModelForCausalLM.from_pretrained(model_name, low_cpu_mem_usage=True)