Spaces:
Sleeping
Sleeping
Commit ·
7a489b1
1
Parent(s): 4246f90
Model prompt change
Browse files
app.py
CHANGED
|
@@ -36,7 +36,7 @@ pipe = pipeline(
|
|
| 36 |
model=chat_model,
|
| 37 |
tokenizer=tokenizer,
|
| 38 |
device_map="auto",
|
| 39 |
-
max_new_tokens=
|
| 40 |
)
|
| 41 |
|
| 42 |
def clean_chat_output(full_text):
|
|
@@ -56,7 +56,7 @@ def explain_recycling(class_label):
|
|
| 56 |
"content": (
|
| 57 |
"You are an expert in waste sorting. "
|
| 58 |
"You ALWAYS answer using exactly two bullet points:\n"
|
| 59 |
-
"• Recycling type: <
|
| 60 |
"• Disposal: <clear, detailed correct sentence>\n"
|
| 61 |
"No extra text, no introductions, no explanations."
|
| 62 |
)
|
|
@@ -75,7 +75,7 @@ def explain_recycling(class_label):
|
|
| 75 |
|
| 76 |
outputs = pipe(
|
| 77 |
prompt,
|
| 78 |
-
max_new_tokens=
|
| 79 |
do_sample=False
|
| 80 |
)
|
| 81 |
|
|
|
|
| 36 |
model=chat_model,
|
| 37 |
tokenizer=tokenizer,
|
| 38 |
device_map="auto",
|
| 39 |
+
max_new_tokens=80
|
| 40 |
)
|
| 41 |
|
| 42 |
def clean_chat_output(full_text):
|
|
|
|
| 56 |
"content": (
|
| 57 |
"You are an expert in waste sorting. "
|
| 58 |
"You ALWAYS answer using exactly two bullet points:\n"
|
| 59 |
+
"• Recycling type: <Item category>\n"
|
| 60 |
"• Disposal: <clear, detailed correct sentence>\n"
|
| 61 |
"No extra text, no introductions, no explanations."
|
| 62 |
)
|
|
|
|
| 75 |
|
| 76 |
outputs = pipe(
|
| 77 |
prompt,
|
| 78 |
+
max_new_tokens=80,
|
| 79 |
do_sample=False
|
| 80 |
)
|
| 81 |
|