Update app.py

app.py

@@ -44,14 +44,14 @@ def pred(text_in, d_model, n_layer, vocab_size, genlen, temperature, top_k, top_
 demo = gr.Interface(
     fn=pred,
     inputs=[
-        gr.inputs.Textbox(label="Input Text"),
-        gr.inputs.Slider(minimum=128, maximum=1024, default=512, label="Model Dimension (d_model)"),
-        gr.inputs.Slider(minimum=1, maximum=24, default=12, label="Number of Layers (n_layer)"),
-        gr.inputs.Number(default=50257, label="Vocabulary Size (vocab_size)"),
-        gr.inputs.Slider(minimum=50, maximum=500, default=200, label="Generation Length (genlen)"),
-        gr.inputs.Slider(minimum=0.1, maximum=1.0, default=0.5, label="Temperature"),
-        gr.inputs.Slider(minimum=1, maximum=50, default=10, label="Top K"),
-        gr.inputs.Slider(minimum=0.1, maximum=1.0, default=0.9, label="Top P"),
+        gr.Textbox(label="Input Text"),
+        gr.Slider(minimum=128, maximum=1024, default=512, label="Model Dimension (d_model)"),
+        gr.Slider(minimum=1, maximum=24, default=12, label="Number of Layers (n_layer)"),
+        gr.Number(default=50257, label="Vocabulary Size (vocab_size)"),
+        gr.Slider(minimum=50, maximum=500, default=200, label="Generation Length (genlen)"),
+        gr.Slider(minimum=0.1, maximum=1.0, default=0.5, label="Temperature"),
+        gr.Slider(minimum=1, maximum=50, default=10, label="Top K"),
+        gr.Slider(minimum=0.1, maximum=1.0, default=0.9, label="Top P"),
     ],
     outputs="text",
     title="Welcome to Tonic's 🐍Mamba",