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'])
|