Update app.py

app.py

@@ -54,9 +54,10 @@ def translate(
     source_text: str, 
     source_lang: str,
     target_lang: str, 
-    max_chunk: int,
     max_length: int,
-    temperature: float):
+    temperature: float,
+    top_p: float,
+    rp: float):
     
     print(f'Text is - {source_text}')
     
@@ -107,7 +108,7 @@ with gr.Blocks(theme="soft", css=CSS) as demo:
                 value="Spanish",
             )
             max_length = gr.Slider(
-                label="Context Window",
+                label="Max Length",
                 minimum=512,
                 maximum=8192,
                 value=4096,
@@ -120,6 +121,20 @@ with gr.Blocks(theme="soft", css=CSS) as demo:
                 value=0.3,
                 step=0.1,
             )
+            top_p = gr.Slider(
+                minimum=0.0,
+                maximum=1.0,
+                step=0.1,
+                value=1.0,
+                label="top_p",
+            )
+            rp = gr.Slider(
+                minimum=0.0,
+                maximum=2.0,
+                step=0.1,
+                value=1.2,
+                label="Repetition penalty",
+            )
             gr.Markdown(DESCRIPTION)
         with gr.Column(scale=4):
             source_text = gr.Textbox(
@@ -138,7 +153,7 @@ with gr.Blocks(theme="soft", css=CSS) as demo:
         clear = gr.ClearButton([source_text, output_text])
         
     source_text.change(lang_detector, source_text, source_lang)
-    submit.click(fn=translate, inputs=[source_text, source_lang, target_lang, max_chunk, max_length, temperature], outputs=[output_text])
+    submit.click(fn=translate, inputs=[source_text, source_lang, target_lang, max_length, temperature, top_p, rp], outputs=[output_text])
 
 
 if __name__ == "__main__":