geethareddy commited on
Commit
39d22d4
·
verified ·
1 Parent(s): 159fa92

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -17,15 +17,17 @@ model = AutoModelForCausalLM.from_pretrained(model_name)
17
  tokenizer.pad_token = tokenizer.eos_token
18
  model.config.pad_token_id = tokenizer.eos_token_id
19
 
20
- # Define a prompt template (no examples, just structured prompt)
21
  PROMPT_TEMPLATE = """You are an AI coach for construction supervisors. Based on the following inputs, generate a daily checklist, focus suggestions, and a motivational quote. Format your response with clear labels as follows:
22
 
23
  Checklist:
24
  - Item 1
25
  - Item 2
 
26
  Suggestions:
27
  - Suggestion 1
28
  - Suggestion 2
 
29
  Quote:
30
  - Your motivational quote here
31
 
 
17
  tokenizer.pad_token = tokenizer.eos_token
18
  model.config.pad_token_id = tokenizer.eos_token_id
19
 
20
+ # Define a prompt template (structured format)
21
  PROMPT_TEMPLATE = """You are an AI coach for construction supervisors. Based on the following inputs, generate a daily checklist, focus suggestions, and a motivational quote. Format your response with clear labels as follows:
22
 
23
  Checklist:
24
  - Item 1
25
  - Item 2
26
+
27
  Suggestions:
28
  - Suggestion 1
29
  - Suggestion 2
30
+
31
  Quote:
32
  - Your motivational quote here
33