tranquilkd commited on
Commit
f1a8924
·
1 Parent(s): 11543bd

corrected app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -100,11 +100,11 @@ iface = gr.Interface(
100
  fn=gradio_interface,
101
  inputs=[
102
  gr.Textbox(label="Prompt", placeholder="Enter your prompt here..."),
103
- gr.Slider(minimum=10, maximum=500, label="Min Length"),
104
- gr.Slider(minimum=10, maximum=500, label="Max Length"),
105
- gr.Slider(minimum=0.1, maximum=2.0, label="Temperature"),
106
- gr.Slider(minimum=1, maximum=100, label="Top K"),
107
- gr.Slider(minimum=0.1, maximum=1.0, label="Top P")
108
  ],
109
  outputs=gr.Textbox(label="Generated Text"),
110
  title="SmolLM2-135M Text Generation",
 
100
  fn=gradio_interface,
101
  inputs=[
102
  gr.Textbox(label="Prompt", placeholder="Enter your prompt here..."),
103
+ gr.Slider(minimum=10, maximum=500, value=28, label="Min Length"),
104
+ gr.Slider(minimum=10, maximum=500, value=64, label="Max Length"),
105
+ gr.Slider(minimum=0.1, maximum=1.0, value=0.8, label="Temperature"),
106
+ gr.Slider(minimum=1, maximum=100, value=50, label="Top K"),
107
+ gr.Slider(minimum=0.1, maximum=1.0, value=0.7, label="Top P")
108
  ],
109
  outputs=gr.Textbox(label="Generated Text"),
110
  title="SmolLM2-135M Text Generation",