Update models.py

models.py

@@ -20,7 +20,7 @@ class Models():
         
         ner_model = AutoModelForCausalLM.from_pretrained(checkpoint, device_map="auto", torch_dtype=torch.float16, offload_folder="offload", offload_state_dict = True)
         tokenizer = AutoTokenizer.from_pretrained("Universal-NER/UniNER-7B-all", use_fast=False, padding="max_length")
-        pipeline = pipeline(
+        hf_pipeline = pipeline(
             "text-generation", #task
             model=ner_model,
             max_length=1000,
@@ -31,7 +31,7 @@ class Models():
             num_return_sequences=1
         )
     
-        self.llm = HuggingFacePipeline(pipeline = pipeline, model_kwargs = {'temperature':0})
+        self.llm = HuggingFacePipeline(hf_pipeline, model_kwargs = {'temperature':0})
         self.prompt = PromptTemplate(template=self.template, input_variables=["input_text","entity_type"])
         self.llm_chain = LLMChain(prompt=self.prompt, llm=self.llm)