prithivMLmods commited on
Commit
6cb637b
1 Parent(s): cfbffd9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -17
app.py CHANGED
@@ -107,26 +107,19 @@ def respond(
107
  response += token
108
  yield response
109
 
110
- # Define individual components
111
- mood_choice = gr.Dropdown(choices=list(mood_prompts.keys()), label="Mood", value="Casual")
112
- system_message = gr.Textbox(value="", label="System message", visible=False)
113
- max_tokens = gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens", visible=False)
114
- temperature = gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature", visible=False)
115
- top_p = gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-P", visible=False)
116
-
117
- # Integrate components into the interface
118
- demo = gr.ChatInterface(
119
- respond,
120
- additional_inputs=[
121
- system_message,
122
- max_tokens,
123
- temperature,
124
- top_p,
125
  ],
126
- interactive=True,
127
  css=css,
128
  theme="allenai/gradio-theme",
129
- secondary_inputs=[mood_choice], # Adding mood_choice separately for visibility
130
  )
131
 
132
  if __name__ == "__main__":
 
107
  response += token
108
  yield response
109
 
110
+ demo = gr.Interface(
111
+ fn=respond,
112
+ inputs=[
113
+ gr.Textbox(label="Input message"),
114
+ gr.Textbox(value="", label="System message", visible=False),
115
+ gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens", visible=False),
116
+ gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature", visible=False),
117
+ gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-P", visible=False),
118
+ gr.Dropdown(choices=list(mood_prompts.keys()), label="Mood", value="Casual"),
 
 
 
 
 
 
119
  ],
120
+ outputs=gr.Textbox(label="Response"),
121
  css=css,
122
  theme="allenai/gradio-theme",
 
123
  )
124
 
125
  if __name__ == "__main__":