Update app.py

app.py

@@ -39,23 +39,22 @@ def pred(text_in, d_model, n_layer, vocab_size, genlen, temperature, top_k, top_
 
     text_out = tokenizer.batch_decode(output.sequences.tolist(), skip_special_tokens=True)
     return text_out[0]
-
-# Define Gradio interface
+    pass
 demo = gr.Interface(
     fn=pred,
     inputs=[
         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"),
+        gr.Slider(minimum=128, maximum=1024, value=512, label="Model Dimension (d_model)"),
+        gr.Slider(minimum=1, maximum=24, value=12, label="Number of Layers (n_layer)"),
+        gr.Number(value=50257, label="Vocabulary Size (vocab_size)"),
+        gr.Slider(minimum=50, maximum=500, value=200, label="Generation Length (genlen)"),
+        gr.Slider(minimum=0.1, maximum=1.0, value=0.5, label="Temperature"),
+        gr.Slider(minimum=1, maximum=50, value=10, label="Top K"),
+        gr.Slider(minimum=0.1, maximum=1.0, value=0.9, label="Top P"),
     ],
     outputs="text",
     title="Welcome to Tonic's 🐍Mamba",
-    description="With this Demo, you can customize the model hyperparameters of [🐍Mamba](https://www.huggingface.co/state-spaces/mamba-2.8b) . Everytime you send a request it will instantiate the model accordingly, so please be patient."
+    description="With this Demo, you can customize the model hyperparameters of [🐍Mamba](https://www.huggingface.co/state-spaces/mamba-2.8b). Every time you send a request it will instantiate the model accordingly, so please be patient."
 )
 
 if __name__ == "__main__":