Update main.py

main.py

@@ -8,7 +8,7 @@ quants = BitsAndBytesConfig(load_in_4bit=True)
 model_id = "mistralai/Mistral-7B-Instruct-v0.2"
 tokenizer = AutoTokenizer.from_pretrained(model_id,quantization_config=quants)
 model = AutoModelForCausalLM.from_pretrained(model_id,quantization_config=quants)
-pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
+pipe = pipeline("text-generation", model=model, tokenizer=tokenizer,max_new_tokens=1000)
 hf = HuggingFacePipeline(pipeline=pipe)
 
 def generate_blog(role , words , topic):