Update app.py

app.py

@@ -41,9 +41,9 @@ def format_prompt(message, history, cust_p):
     prompt+=cust_p.replace("USER_INPUT",message)
     return prompt
 
-def chat_inf(system_prompt,prompt,history,memory,client_choice,seed,temp,tokens,top_p,rep_p,chat_mem,cust_p,translate_fa,translate_prompt):
+def chat_inf(system_prompt,prompt,history,memory,client_choice,seed,temp,tokens,top_p,rep_p,chat_mem,cust_p,translate_fa):
     #token max=8192
-    if(translate_prompt == True):
+    if(translate_fa == True):
         if(len(prompt) > 2000):
             translatedtext1 = GoogleTranslator(source='auto', target='en').translate(prompt[0:2000])
             translatedtext2 = GoogleTranslator(source='auto', target='en').translate(prompt[2000:(len(prompt))])
@@ -131,8 +131,7 @@ with gr.Blocks() as app:
                     custom_prompt=gr.Textbox(label="Modify Prompt Format", info="For testing purposes. 'USER_INPUT' is where 'SYSTEM_PROMPT, PROMPT' will be placed", lines=5,value="<start_of_turn>userUSER_INPUT<end_of_turn><start_of_turn>model")
             with gr.Column(scale=1):
                 with gr.Group():
-                    translate_fa = gr.Checkbox(label="Translate Response to Persian", value=True)
-                    translate_prompt = gr.Checkbox(label="Translate Prompt to English", value=True)
+                    translate_fa = gr.Checkbox(label="Translate to Persian", value=True)
                     rand = gr.Checkbox(label="Random Seed", value=True)
                     seed=gr.Slider(label="Seed", minimum=1, maximum=1111111111111111,step=1, value=rand_val)
                     tokens = gr.Slider(label="Max new tokens",value=1600,minimum=0,maximum=8000,step=64,interactive=True, visible=True,info="The maximum number of tokens")
@@ -145,8 +144,8 @@ with gr.Blocks() as app:
     client_choice.change(load_models,client_choice,[chat_b])
     app.load(load_models,client_choice,[chat_b])
     
-    chat_sub=inp.submit(check_rand,[rand,seed],seed).then(chat_inf,[sys_inp,inp,chat_b,memory,client_choice,seed,temp,tokens,top_p,rep_p,chat_mem,custom_prompt,translate_fa,translate_prompt],[chat_b,memory])
-    go=btn.click(check_rand,[rand,seed],seed).then(chat_inf,[sys_inp,inp,chat_b,memory,client_choice,seed,temp,tokens,top_p,rep_p,chat_mem,custom_prompt,translate_fa,translate_prompt],[chat_b,memory])
+    chat_sub=inp.submit(check_rand,[rand,seed],seed).then(chat_inf,[sys_inp,inp,chat_b,memory,client_choice,seed,temp,tokens,top_p,rep_p,chat_mem,custom_prompt,translate_fa],[chat_b,memory])
+    go=btn.click(check_rand,[rand,seed],seed).then(chat_inf,[sys_inp,inp,chat_b,memory,client_choice,seed,temp,tokens,top_p,rep_p,chat_mem,custom_prompt,translate_fa],[chat_b,memory])
 
     clear_btn.click(clear_fn,None,[inp,sys_inp,chat_b,memory])
 app.queue(default_concurrency_limit=10).launch()