Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -28,7 +28,7 @@ def process(text: str,
|
|
28 |
"top_k": top_k,
|
29 |
"top_p": top_p,
|
30 |
"temperature": temp,
|
31 |
-
"num_return_sequences":
|
32 |
"repetition_penalty": 2.0,
|
33 |
},
|
34 |
"options": {
|
@@ -97,7 +97,6 @@ if st.button("Run"):
|
|
97 |
for error in result["error"]:
|
98 |
st.write(f'{error}')
|
99 |
else:
|
100 |
-
|
101 |
-
|
102 |
-
|
103 |
-
st.write(result)
|
|
|
28 |
"top_k": top_k,
|
29 |
"top_p": top_p,
|
30 |
"temperature": temp,
|
31 |
+
"num_return_sequences": num_return_sequences,
|
32 |
"repetition_penalty": 2.0,
|
33 |
},
|
34 |
"options": {
|
|
|
97 |
for error in result["error"]:
|
98 |
st.write(f'{error}')
|
99 |
else:
|
100 |
+
for i in range(num_return_sequences):
|
101 |
+
result = result[i]["generated_text"]
|
102 |
+
st.write(result.replace("\n", " \n"))
|
|