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 |
|