Improved Prompts

run.py

@@ -13,9 +13,9 @@ def give(results):
   if int(results.pred[0][-1][-1].numpy())==0:
       prompt="Great focus. You are doing great!"
   elif int(results.pred[0][-1][-1].numpy())==1:
-      prompt="You look dizzy."
+      prompt="Suggest an exercise for staying awake."
   else:
-      prompt="You are getting distracted."
+      prompt="Suggest an exercise for staying alert."
   inputs = tokenizer(prompt, return_tensors="pt", return_attention_mask=False)
   outputs = llm.generate(**inputs, max_length=30)
   return tokenizer.batch_decode(outputs)[0]