Update README.md

README.md

@@ -14,7 +14,7 @@ widget:
 
 inference:
   parameters:
-    max_length: 200
+    max_length: 80
 ---
 
 # GPT-2
@@ -31,13 +31,9 @@ model = GPTNeoForCausalLM.from_pretrained('minhtoan/gpt3-small-vietnamese')
 
 text = "Hoa quả và rau thường rẻ hơn khi vào mùa"
 input_ids = tokenizer.encode(text, return_tensors='pt')
-max_length = 150
+max_length = 80
 
-sample_outputs = model.generate(input_ids,pad_token_id=tokenizer.eos_token_id,
-                                   do_sample=True,
-                                   max_length=max_length,
-                                   min_length=max_length,
-                                   num_return_sequences=1)
+sample_outputs = model.generate(input_ids, do_sample=True, max_length=max_length)
 
 for i, sample_output in enumerate(sample_outputs):
     print(">> Generated text {}\n\n{}".format(i+1, tokenizer.decode(sample_output.tolist())))