Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ def load_model():
|
|
16 |
return model
|
17 |
|
18 |
def get_response(
|
19 |
-
input_text, num_return_sequences, num_beams, max_length=
|
20 |
):
|
21 |
|
22 |
model = load_model()
|
@@ -31,7 +31,7 @@ context = st.text_area(label="Write or paste text", max_chars=512)
|
|
31 |
|
32 |
with st.expander("Advanced"):
|
33 |
temperature = st.slider("Temperature", 0.1, 5.0, 1.5, 0.1)
|
34 |
-
max_length = st.slider("Max length", 10,
|
35 |
|
36 |
if context:
|
37 |
response = get_response(context, num_return_sequences, num_beams, max_length, temperature)
|
|
|
16 |
return model
|
17 |
|
18 |
def get_response(
|
19 |
+
input_text, num_return_sequences, num_beams, max_length=512, temperature=1.5
|
20 |
):
|
21 |
|
22 |
model = load_model()
|
|
|
31 |
|
32 |
with st.expander("Advanced"):
|
33 |
temperature = st.slider("Temperature", 0.1, 5.0, 1.5, 0.1)
|
34 |
+
max_length = st.slider("Max length", 10, 512, 256, 10)
|
35 |
|
36 |
if context:
|
37 |
response = get_response(context, num_return_sequences, num_beams, max_length, temperature)
|