Spaces:
Running
Running
ayan4m1
commited on
Commit
·
6b3b146
1
Parent(s):
97b4c79
fix: yield correct value
Browse files
app.py
CHANGED
@@ -12,19 +12,14 @@ def respond(
|
|
12 |
top_p: float,
|
13 |
top_k: int
|
14 |
):
|
15 |
-
yield
|
16 |
-
|
17 |
-
|
18 |
-
|
19 |
-
|
20 |
-
|
21 |
-
|
22 |
-
|
23 |
-
top_p=top_p,
|
24 |
-
top_k=top_k
|
25 |
-
)['generated_text']
|
26 |
-
}
|
27 |
-
]
|
28 |
|
29 |
|
30 |
"""
|
|
|
12 |
top_p: float,
|
13 |
top_k: int
|
14 |
):
|
15 |
+
yield pipe(
|
16 |
+
message,
|
17 |
+
max_new_tokens=max_new_tokens,
|
18 |
+
do_sample=True,
|
19 |
+
temperature=temperature,
|
20 |
+
top_p=top_p,
|
21 |
+
top_k=top_k
|
22 |
+
)[0]['generated_text']
|
|
|
|
|
|
|
|
|
|
|
23 |
|
24 |
|
25 |
"""
|