karimaloulou commited on
Commit
f4c58dc
·
verified ·
1 Parent(s): 6c5d815

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -59,9 +59,9 @@ For information on how to customize the ChatInterface, peruse the gradio docs: h
59
  demo = gr.ChatInterface(
60
  respond,
61
  additional_inputs=[
62
- gr.Textbox(value=f"""<s>[INST] Given these TTPs: {techniques_str}\n\nfigure out which technique is used in these logs and respond in bullets points and nothing else: [/INST]""", label="System message"),
63
- gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
64
- gr.Slider(minimum=0.1, maximum=1.0, value=0.7, step=0.1, label="Temperature"),
65
  gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"),
66
  ],
67
  )
 
59
  demo = gr.ChatInterface(
60
  respond,
61
  additional_inputs=[
62
+ gr.Textbox(value=f"""<s>[INST] Given these TTPs: {techniques_str}\n\nfigure out which technique is used in these logs and respond in bullets points and nothing else if there is no TTPs detected just say that there is not TTP detected: [/INST]""", label="System message"),
63
+ gr.Slider(minimum=1, maximum=2048, value=1024, step=1, label="Max new tokens"),
64
+ gr.Slider(minimum=0.1, maximum=1.0, value=0.1, step=0.1, label="Temperature"),
65
  gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"),
66
  ],
67
  )