Spaces:
Running
Running
Update app.py
Browse filesKey Features:
• System Message: Tailored for bug bounty and cybersecurity tasks.
• Interactive UI: Allow users to adjust parameters like token limits, temperature, and top-p for the AI model.
• Gradio Interface: A simple, customizable web interface that makes it easy to interact with the model.
app.py
CHANGED
@@ -46,7 +46,7 @@ For information on how to customize the ChatInterface, peruse the gradio docs: h
|
|
46 |
demo = gr.ChatInterface(
|
47 |
respond,
|
48 |
additional_inputs=[
|
49 |
-
gr.Textbox(value="You are a
|
50 |
gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
|
51 |
gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
|
52 |
gr.Slider(
|
@@ -59,6 +59,4 @@ demo = gr.ChatInterface(
|
|
59 |
],
|
60 |
)
|
61 |
|
62 |
-
|
63 |
-
if __name__ == "__main__":
|
64 |
-
demo.launch()
|
|
|
46 |
demo = gr.ChatInterface(
|
47 |
respond,
|
48 |
additional_inputs=[
|
49 |
+
gr.Textbox(value="You are a chatbot trained to assist with specific tasks in bug bounty hunting, offering advice on penetration testing, vulnerability discovery, and other security-related topics.", label="System message"),
|
50 |
gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
|
51 |
gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
|
52 |
gr.Slider(
|
|
|
59 |
],
|
60 |
)
|
61 |
|
62 |
+
demo.launch()
|
|
|
|