breadlicker45 commited on
Commit
650f659
1 Parent(s): fe8a926

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -20,7 +20,7 @@ with st.sidebar:
20
 
21
  num_beams = st.slider('N° tree beams search', 1, 15, 1)
22
 
23
- temp = st.slider('temperature', 0, 1, 1, 0.1)
24
 
25
  early_stopping = st.selectbox(
26
  'Early stopping text generation',
@@ -58,7 +58,7 @@ with col2:
58
  gen = st.info('Generating text...')
59
  answer = generator(prompt,
60
  max_length=max_length, no_repeat_ngram_size=no_ngram_repeat,
61
- early_stopping=early_stopping, num_beams=num_beams, min_length=min_length, temperature=temp)
62
  gen.empty()
63
 
64
  lst = answer[0]['generated_text']
 
20
 
21
  num_beams = st.slider('N° tree beams search', 1, 15, 1)
22
 
23
+
24
 
25
  early_stopping = st.selectbox(
26
  'Early stopping text generation',
 
58
  gen = st.info('Generating text...')
59
  answer = generator(prompt,
60
  max_length=max_length, no_repeat_ngram_size=no_ngram_repeat,
61
+ early_stopping=early_stopping, num_beams=num_beams, min_length=min_length)
62
  gen.empty()
63
 
64
  lst = answer[0]['generated_text']