Update src/agents/product_manager_agent.py

src/agents/product_manager_agent.py

@@ -1,13 +1,12 @@
 from transformers import LlamaTokenizer, AutoModelForCausalLM
-from transformers import AutoTokenizer
 import torch
 
 MODEL_REPO = "Rahul-8799/product_manager_mistral"
 
-tokenizer = AutoTokenizer.from_pretrained("Rahul-8799/product_manager_mistral")
-
-#tokenizer = LlamaTokenizer.from_pretrained(MODEL_REPO)
-model = AutoModelForCausalLM.from_pretrained(MODEL_REPO, device_map="auto", torch_dtype=torch.float16)
+tokenizer = LlamaTokenizer.from_pretrained(MODEL_REPO)
+model = AutoModelForCausalLM.from_pretrained(
+    MODEL_REPO, device_map="auto", torch_dtype=torch.float16
+)
 model.eval()
 
 def run(prompt):