add min step

app.py

@@ -22,8 +22,7 @@ generator = pipeline('text-generation', model=model, tokenizer='bert-base-chines
 
 def generate(prompt, num_beams, max_length, repetition_penalty, seed):
 
-    if seed is not None:
-        torch.manual_seed(seed)
+    torch.manual_seed(seed)
 
     outputs = generator(prompt, max_length=max_length, num_return_sequences=5, num_beams=num_beams, repetition_penalty=repetition_penalty)
     output_texts = [output['generated_text'] for output in outputs]
@@ -37,15 +36,15 @@ examples = [["子曰", 10, 50, 1.5, 42],
             ["子路问仁", 10, 50, 1.5, 42],
             ["孙行者笑道", 10, 50, 1.5, 42],
             ["牛魔王与红孩儿", 10, 50, 1.5, 42],
-            ["鲲鹏", 10, 50, 1.5, 42], 
+            ["鲲鹏", 10, 50, 1.5, 42],
             ["宝玉道", 10, 50, 1.5, 42],
             ["黛玉行至贾母处", 10, 50, 1.5, 42],]
 
 
 iface = gr.Interface(fn=generate,
                      inputs=[gr.inputs.Textbox(lines=2, label="Prompt"),
-                             gr.inputs.Slider(minimum=1, maximum=20, default=10, label="Number of beams"),
-                              gr.inputs.Slider(minimum=10, maximum=100, default=50, label="Max length"),
+                             gr.inputs.Slider(minimum=1, maximum=20, default=10, step=1, label="Number of beams"),
+                              gr.inputs.Slider(minimum=10, maximum=100, default=50, step=1, label="Max length"),
                               gr.inputs.Slider(minimum=1, maximum=5, default=1.5, label="Repetition penalty"),
                               gr.inputs.Number(default=0, label="Seed")],
                      outputs=gr.outputs.Textbox(label="Generated Text"),