Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 (
|
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 |
|