amirgame197 commited on
Commit
aa621e3
·
verified ·
1 Parent(s): 2131773

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,7 +16,7 @@ def format_prompt(message, history):
16
  return prompt
17
 
18
  def generate(
19
- prompt, history, system_prompt, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,translate_fa,
20
  ):
21
  if(translate_fa == True):
22
  if(len(prompt) > 2000):
@@ -52,7 +52,7 @@ def generate(
52
 
53
 
54
  additional_inputs=[
55
- translate_fa = gr.Checkbox(
56
  label="Translate to Persian",
57
  value=True
58
  ),
 
16
  return prompt
17
 
18
  def generate(
19
+ prompt, history, translate_fa, system_prompt, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,
20
  ):
21
  if(translate_fa == True):
22
  if(len(prompt) > 2000):
 
52
 
53
 
54
  additional_inputs=[
55
+ gr.Checkbox(
56
  label="Translate to Persian",
57
  value=True
58
  ),