Canstralian commited on
Commit
95eb2b4
·
verified ·
1 Parent(s): dfb8e28

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +36 -12
app.py CHANGED
@@ -41,24 +41,48 @@ def respond(
41
 
42
 
43
  """
44
- For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
 
 
 
 
 
 
 
 
 
 
45
  """
46
  demo = gr.ChatInterface(
47
  respond,
48
  additional_inputs=[
49
- gr.Textbox(value="You are a friendly Chatbot.", 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(
53
- minimum=0.1,
54
- maximum=1.0,
55
- value=0.95,
56
- step=0.05,
57
- label="Top-p (nucleus sampling)",
 
 
 
 
 
 
 
 
 
 
 
 
 
 
58
  ),
59
  ],
60
  )
61
 
62
-
63
- if __name__ == "__main__":
64
- demo.launch()
 
41
 
42
 
43
  """
44
+ System Prompt Modification for NLPToolkit Agent
45
+ """
46
+ default_system_message = (
47
+ "You are NLPToolkit Agent, an advanced natural language processing assistant. "
48
+ "You specialize in tasks such as text summarization, sentiment analysis, text classification, "
49
+ "entity recognition, and answering technical questions about NLP models and datasets. "
50
+ "Assist users with clear, concise, and actionable outputs."
51
+ )
52
+
53
+ """
54
+ Updated Gradio Interface
55
  """
56
  demo = gr.ChatInterface(
57
  respond,
58
  additional_inputs=[
59
+ gr.Textbox(
60
+ value=default_system_message,
61
+ label="System message"
62
+ ),
63
  gr.Slider(
64
+ minimum=1,
65
+ maximum=2048,
66
+ value=512,
67
+ step=1,
68
+ label="Max new tokens"
69
+ ),
70
+ gr.Slider(
71
+ minimum=0.1,
72
+ maximum=4.0,
73
+ value=0.7,
74
+ step=0.1,
75
+ label="Temperature"
76
+ ),
77
+ gr.Slider(
78
+ minimum=0.1,
79
+ maximum=1.0,
80
+ value=0.95,
81
+ step=0.05,
82
+ label="Top-p (nucleus sampling)"
83
  ),
84
  ],
85
  )
86
 
87
+ # Run the app
88
+ demo.launch()