Wootang01 commited on
Commit
bb4e2f1
1 Parent(s): 1954bdd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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=128, temperature=1.5
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, 100, 60, 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)