Update app.py

app.py

@@ -23,7 +23,8 @@ def temperature_sampling(logits, temperature):
     sampled_token = torch.multinomial(probabilities, 1)
     return sampled_token
 
-def CTXGen(τ, g_num, start, end):
+def CTXGen(τ, g_num, length_range):
+    start, end = length_range
     X1 = "X"
     X2 = "X"
     X4 = ""
@@ -120,8 +121,7 @@ iface = gr.Interface(
     inputs=[
         gr.Slider(minimum=1, maximum=2, step=0.01, label="τ"),
         gr.Dropdown(choices=[1,10,100], label="Number of generations"),
-        gr.Textbox(label="Min length"),
-        gr.Textbox(label="Max length")
+        gr.RangeSlider(minimum=8, maximum=50, step=1, label="Length Range (Min, Max)")
     ],
     outputs=["file"]
 )