Update app.py

app.py

@@ -18,7 +18,7 @@ def gera_texto(start, temperature, max_new_tokens, num_samples):
                   temperature=temperature,
                   top_p=0.5,
                   nsamples=num_samples,
-                  batch_size=5,
+                  batch_size= 2,
                   return_as_list=True
                   )
     k =0
@@ -29,13 +29,13 @@ def gera_texto(start, temperature, max_new_tokens, num_samples):
 with st.form("my_form"):
     col1, col2, col3 = st.columns(3)
     with col1:
-        int_samples = st.slider('Exemplos', min_value=1, max_value=10, value=5, step=1)
+        int_samples = st.slider('Exemplos', min_value=1, max_value=10, value=4, step=2)
     with col2:
         int_size = st.slider('Num Tokens', min_value=20, max_value=500, value=160, step=5)
     with col3:
         int_temp = st.number_input("Temperatura",min_value=0.1,max_value=2.0,value=1.2,step=0.1,format="%.1f")
 
-    source = st.text_area("Escolha uma frase inicial", value="Quero voce ao meu lado", placeholder="Entre com o inicio da musica...")
+    source = st.text_area("Escolha uma frase inicial", value="Contrui uma casa para nos", placeholder="Entre com o inicio da musica...")
 
     submitted = st.form_submit_button("Gerar músicas")
     if submitted: