Update README.md
Browse files
README.md
CHANGED
@@ -93,7 +93,7 @@ tokenizer.add_eos_token = True
|
|
93 |
tokenizer.add_bos_token, tokenizer.add_eos_token
|
94 |
|
95 |
|
96 |
-
pipe = pipeline(task="text-generation", model=
|
97 |
prompt = "Enter your query"
|
98 |
result = pipe(prompt)
|
99 |
print(result[0]['generated_text'])
|
|
|
93 |
tokenizer.add_bos_token, tokenizer.add_eos_token
|
94 |
|
95 |
|
96 |
+
pipe = pipeline(task="text-generation", model=model_id, tokenizer=tokenizer, max_length=600)
|
97 |
prompt = "Enter your query"
|
98 |
result = pipe(prompt)
|
99 |
print(result[0]['generated_text'])
|