Update app.py

app.py

@@ -47,9 +47,12 @@ with col1:
     )
 
 with col2:
-    temperature = st.number_input('Temperature: ', min_value=0.0, max_value=1.0, value=0.5, format='%f')
+    temperature = st.slider('Temperature: ', min_value=0.0, max_value=1.0, value=0.5)
+    top_p = st.slider('Top p: ', min_value=0.5, max_value=1.0, value=0.99)
     max_tokens = st.number_input('Max output length: ', min_value=1, max_value=64, format='%i')
-
+    threshold = st.number_input('Min token score:: ', value=-np.inf)
+    
+    
 tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
 
 model = T5ForConditionalGeneration.from_pretrained(  
@@ -99,5 +102,5 @@ if instruction:
     
     st.write(output_text)
 
-    fig = plot_word_scores(top_token_ids(outputs, threshold=-1), outputs, tokenizer)
-    st.plotly_chart(fig, use_container_width=False)
+    fig = plot_word_scores(top_token_ids(outputs, threshold=threshold), outputs, tokenizer)
+    st.plotly_chart(fig, theme=None, use_container_width=False)