[ADD] Temperature parameter

app.py

@@ -36,14 +36,15 @@ def remove_empty_lines(text):
     non_empty_lines = [line for line in lines if line.strip()]
     return "\n".join(non_empty_lines)
 
-def predict(prompt, max_length, repetition_penalty=1.3):
+def predict(prompt, max_length, repetition_penalty, temperature):
     print("Dentro da xeración...")
     prompt_length = len(generator_model.tokenizer.encode(prompt))
     generated_text = generator_model(
         prompt,
         max_length=prompt_length + max_length, 
         pad_token_id=generator_model.tokenizer.eos_token_id, 
-        repetition_penalty=repetition_penalty)
+        repetition_penalty=repetition_penalty,
+        temperature=temperature)
 
     generated_sequence = generated_text[0]['generated_text']
     if generated_sequence is  None:
@@ -60,7 +61,8 @@ def clear():
         None, 
         None,
         gr.update(value=20),
-        gr.update(value=1.3)
+        gr.update(value=1.3),
+        gr.update(value=0.5)
     )
 def pass_to_input(generated_gl):
     return (
@@ -71,13 +73,15 @@ def pass_to_input(generated_gl):
 def parameters_default(text):
     return (
         gr.update(value=30), # max_length
-        gr.update(value=1.3) # repetition_penalty
+        gr.update(value=1.3), # repetition_penalty
+        gr.update(value=0.5) # temperature
     )
 
 def parameters_fewshot_prompt(text):
     return (
         gr.update(value=15), # max_length
-        gr.update(value=1) # repetition_penalty
+        gr.update(value=1), # repetition_penalty
+        gr.update(value=0.5) # temperature
     )
 
 def gradio_app():
@@ -108,6 +112,12 @@ def gradio_app():
                                 value=1.3,
                                 label="Repetition penalty"
                             )
+                            temperature = Slider(
+                                minimum=0,
+                                maximum=1,
+                                value=0.5,
+                                label="Temperaturr"
+                            )
                 generator_btn = gr.Button(value="Generate",variant='primary')
             with gr.Column():
                 generated_gl = gr.Textbox(label="Output", 
@@ -118,8 +128,8 @@ def gradio_app():
                 pass_btn = gr.Button(value="Pass text to input")
                 clean_btn = gr.Button(value="Clean")
 
-        generator_btn.click(predict, inputs=[text_gl,max_length, repetition_penalty], outputs=generated_gl, api_name="generate-flor-gl")
-        clean_btn.click(fn=clear, inputs=[], outputs=[text_gl, generated_gl, max_length, repetition_penalty], queue=False, api_name=False)
+        generator_btn.click(predict, inputs=[text_gl,max_length, repetition_penalty, temperature], outputs=generated_gl, api_name="generate-flor-gl")
+        clean_btn.click(fn=clear, inputs=[], outputs=[text_gl, generated_gl, max_length, repetition_penalty, temperature], queue=False, api_name=False)
         pass_btn.click(fn=pass_to_input, inputs=[generated_gl], outputs=[text_gl,generated_gl], queue=False, api_name=False)
         
         with gr.Row():
@@ -128,7 +138,7 @@ def gradio_app():
                     label = "Short prompts",
                     examples = short_prompts_examples,
                     inputs = [text_gl],
-                    outputs = [max_length, repetition_penalty],
+                    outputs = [max_length, repetition_penalty, temperature],
                     fn = parameters_default,
                     run_on_click = True
                 )
@@ -136,7 +146,7 @@ def gradio_app():
                     label = "Few-shot prompts",
                     examples = few_shot_prompts_examples,
                     inputs = [text_gl],
-                    outputs = [max_length, repetition_penalty],
+                    outputs = [max_length, repetition_penalty, temperature],
                     fn = parameters_fewshot_prompt,
                     run_on_click = True
                 )