Set min length

app.py

@@ -8,8 +8,8 @@ model = AutoModelForCausalLM.from_pretrained("rinna/japanese-gpt-1b")
 
 generator = pipeline("text-generation", tokenizer=tokenizer, model=model)
 
-def generate(text, max_length=512):
-    out = generator(text, do_sample=True, max_length=max_length, num_return_sequences=1)
+def generate(text, min_length=512):
+    out = generator(text, do_sample=True, min_length=min_length, max_length=1024, num_return_sequences=1)
     text = out[0]['generated_text']
 
     return text, translate_from_jp_to_en(text)
@@ -20,7 +20,7 @@ description = "Demo for generating text in Japanase using a GPT model"
 article = "Built by Narrativa"
 examples = [['日本のeスポーツ障害者がステレオタイプを撃ち落とす', 128]]
 gr.Interface(fn=generate, inputs=[gr.inputs.Textbox(lines=4, label="Prompt"),
-                                  gr.inputs.Slider(minimum=8, maximum=1024, step=8, default=64, label="Number of tokens")],
+                                  gr.inputs.Slider(minimum=8, maximum=1024, step=8, default=128, label="Number of tokens")],
              outputs=["text", "text"],
              title=title, description=description,
              article= article,