LLM_based_Engagement_lvl_alert_system

Sleeping

tanishq1508 commited on Oct 21, 2023

Commit

093ec4e

1 Parent(s): 2cd32e1

Improved prompts

Files changed (1) hide show

run.py CHANGED Viewed

@@ -11,7 +11,7 @@ tokenizer = AutoTokenizer.from_pretrained("microsoft/phi-1_5", trust_remote_code
 def give(results):
   prompt=""
   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="Suggest an exercise for staying awake."
   else:

 def give(results):
   prompt=""
   if int(results.pred[0][-1][-1].numpy())==0:
+      prompt="Suggest a statement for praising my focus."
   elif int(results.pred[0][-1][-1].numpy())==1:
       prompt="Suggest an exercise for staying awake."
   else: