Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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()
|