Spaces:
Sleeping
Sleeping
Commit ·
4c4bef2
1
Parent(s): f8bcafb
prompt model change
Browse files
app.py
CHANGED
|
@@ -49,7 +49,7 @@ def explain_recycling(class_label):
|
|
| 49 |
2. **How {class_label} is recycled or processed**
|
| 50 |
3. **Optional tips for reducing waste or reusing {class_label}**
|
| 51 |
|
| 52 |
-
Write the response in
|
| 53 |
"""
|
| 54 |
|
| 55 |
inputs = tokenizer(prompt, return_tensors="pt").input_ids
|
|
@@ -60,7 +60,7 @@ def explain_recycling(class_label):
|
|
| 60 |
do_sample=True,
|
| 61 |
top_p = 0.9,)
|
| 62 |
|
| 63 |
-
return tokenizer.decode(outputs[0]
|
| 64 |
|
| 65 |
# ------------------ PIPELINE ------------------
|
| 66 |
def full_pipeline(image):
|
|
|
|
| 49 |
2. **How {class_label} is recycled or processed**
|
| 50 |
3. **Optional tips for reducing waste or reusing {class_label}**
|
| 51 |
|
| 52 |
+
Write the response in 3 paragraphs.
|
| 53 |
"""
|
| 54 |
|
| 55 |
inputs = tokenizer(prompt, return_tensors="pt").input_ids
|
|
|
|
| 60 |
do_sample=True,
|
| 61 |
top_p = 0.9,)
|
| 62 |
|
| 63 |
+
return tokenizer.decode(outputs[0])
|
| 64 |
|
| 65 |
# ------------------ PIPELINE ------------------
|
| 66 |
def full_pipeline(image):
|