Update app.py

app.py

@@ -19,6 +19,8 @@ with st.sidebar:
     min_length = st.slider('Min text length', 80, 500, 80)
 
     num_beams = st.slider('N° tree beams search', 1, 15,  1)
+
+    temp = st.slider('temperature', 0, 1,  1)
     
     early_stopping = st.selectbox(
      'Early stopping text generation',
@@ -56,7 +58,7 @@ with col2:
 gen = st.info('Generating text...')
 answer = generator(prompt,
                        max_length=max_length, no_repeat_ngram_size=no_ngram_repeat,
-                        early_stopping=early_stopping, num_beams=num_beams, min_length=min_length)                      
+                        early_stopping=early_stopping, num_beams=num_beams, min_length=min_length, temperature=temp)                      
 gen.empty()                      
                        
 lst = answer[0]['generated_text']