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)
|