Spaces:
Sleeping
Sleeping
Commit ·
0bd541d
1
Parent(s): 7a489b1
Model prompt change
Browse files
app.py
CHANGED
|
@@ -76,7 +76,9 @@ def explain_recycling(class_label):
|
|
| 76 |
outputs = pipe(
|
| 77 |
prompt,
|
| 78 |
max_new_tokens=80,
|
| 79 |
-
do_sample=
|
|
|
|
|
|
|
| 80 |
)
|
| 81 |
|
| 82 |
raw = outputs[0]["generated_text"]
|
|
|
|
| 76 |
outputs = pipe(
|
| 77 |
prompt,
|
| 78 |
max_new_tokens=80,
|
| 79 |
+
do_sample=True,
|
| 80 |
+
top_p = 0.9,
|
| 81 |
+
temperature=0.3
|
| 82 |
)
|
| 83 |
|
| 84 |
raw = outputs[0]["generated_text"]
|