Update app.py

app.py

@@ -11,7 +11,7 @@ model = AutoModelForCausalLM.from_pretrained(model_name)
 # Generation function
 def generate_affirmation(description):
     # Structured prompt to guide model output
-    prompt = f"[SUBJECT] learning [/SUBJECT] [STREAK] current performance context [/STREAK] [CONTEXT] {description} [/CONTEXT] [AFFIRMATION]"
+    prompt = f"[SUBJECT] learning [/SUBJECT] [STREAK] current performance context [/STREAK] [CONTEXT] {description} [/CONTEXT] [AFFIRMATION] Give two positive affirmations. [/AFFIRMATION]"
 
     inputs = tokenizer(prompt, return_tensors="pt")
     input_ids = inputs["input_ids"]
@@ -19,7 +19,7 @@ def generate_affirmation(description):
     with torch.no_grad():
         outputs = model.generate(
             input_ids,
-            max_new_tokens=60,
+            max_new_tokens=100,
             temperature=0.7,
             top_k=50,
             top_p=0.95,