Spaces:
Running
Running
Update models.py
Browse files
models.py
CHANGED
@@ -31,7 +31,7 @@ class Models():
|
|
31 |
num_return_sequences=1
|
32 |
)
|
33 |
|
34 |
-
self.llm = HuggingFacePipeline(hf_pipeline, model_kwargs = {'temperature':0})
|
35 |
self.prompt = PromptTemplate(template=self.template, input_variables=["input_text","entity_type"])
|
36 |
self.llm_chain = LLMChain(prompt=self.prompt, llm=self.llm)
|
37 |
|
|
|
31 |
num_return_sequences=1
|
32 |
)
|
33 |
|
34 |
+
self.llm = HuggingFacePipeline(pipeline = hf_pipeline, model_kwargs = {'temperature':0})
|
35 |
self.prompt = PromptTemplate(template=self.template, input_variables=["input_text","entity_type"])
|
36 |
self.llm_chain = LLMChain(prompt=self.prompt, llm=self.llm)
|
37 |
|