ierhon commited on
Commit
e5d1bd4
1 Parent(s): 5807bc7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -77,10 +77,10 @@ def train(message: str, epochs: int, learning_rate: float, emb_size: int, inp_le
77
  return responses[np.argmax(prediction)]
78
 
79
  iface = gr.Interface(fn=train, inputs=["text",
80
- gr.inputs.Slider(1, 64, 32, step=1, label="Epochs"),
81
- gr.inputs.Slider(0.00000001, 0.1, 0.001, step=0.00000001, label="Learning rate"),
82
- gr.inputs.Slider(1, 256, 100, step=1, label="Embedding size"),
83
- gr.inputs.Slider(1,128, step=1, label="Input Length"),
84
  "text"],
85
  outputs="text")
86
  iface.launch()
 
77
  return responses[np.argmax(prediction)]
78
 
79
  iface = gr.Interface(fn=train, inputs=["text",
80
+ gr.inputs.Slider(1, 64, value=32, step=1, label="Epochs"),
81
+ gr.inputs.Slider(0.00000001, 0.1, value=0.001, step=0.00000001, label="Learning rate"),
82
+ gr.inputs.Slider(1, 256, value=100, step=1, label="Embedding size"),
83
+ gr.inputs.Slider(1, 128, value=16, step=1, label="Input Length"),
84
  "text"],
85
  outputs="text")
86
  iface.launch()