ak0601 commited on
Commit
1e84e34
·
verified ·
1 Parent(s): 20ee9b7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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=model, tokenizer=tokenizer, max_length=600)
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'])