hanzla commited on
Commit
ef005a1
·
1 Parent(s): 42ce24e
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -27,7 +27,6 @@ def chat_function(message, history,max_new_tokens,temperature):
27
  max_new_tokens=max_new_tokens,
28
  do_sample=True,
29
  temperature=temp,
30
- top_p=0.9,
31
  )
32
  return outputs[0]["generated_text"][len(prompt):]
33
 
@@ -41,9 +40,9 @@ gr.ChatInterface(
41
  Feel free to play with customization in the "Additional Inputs".
42
  Fine tune Notebook: https://www.kaggle.com/code/hanzlajavaid/gemma-finetuning-raft-technique
43
  """,
44
- theme="Monochrome",
45
  additional_inputs=[
46
- gr.Slider(512, 4096, label="Max New Tokens"),
47
- gr.Slider(0, 1, label="Temperature")
48
  ]
49
  ).launch()
 
27
  max_new_tokens=max_new_tokens,
28
  do_sample=True,
29
  temperature=temp,
 
30
  )
31
  return outputs[0]["generated_text"][len(prompt):]
32
 
 
40
  Feel free to play with customization in the "Additional Inputs".
41
  Fine tune Notebook: https://www.kaggle.com/code/hanzlajavaid/gemma-finetuning-raft-technique
42
  """,
43
+ theme="Soft",
44
  additional_inputs=[
45
+ gr.Slider(512, 4096, value=1024,label="Max New Tokens"),
46
+ gr.Slider(0, 1,value=0.5 ,label="Temperature")
47
  ]
48
  ).launch()