ChatOpenAI
Browse files
app.py
CHANGED
@@ -30,7 +30,7 @@ class LegalExpert:
|
|
30 |
|
31 |
self.huggingface_llm = pipeline("text-generation", model=model_name, tokenizer=AutoTokenizer.from_pretrained(model_name))
|
32 |
|
33 |
-
self.openai_gpt4_llm = ChatOpenAI(
|
34 |
# self.chat = ChatAnthropic()
|
35 |
|
36 |
self.chain = LLMChain(llm=self.openai_gpt4_llm, prompt=full_prompt_template)
|
|
|
30 |
|
31 |
self.huggingface_llm = pipeline("text-generation", model=model_name, tokenizer=AutoTokenizer.from_pretrained(model_name))
|
32 |
|
33 |
+
self.openai_gpt4_llm = ChatOpenAI(temperature=0, max_tokens=256)
|
34 |
# self.chat = ChatAnthropic()
|
35 |
|
36 |
self.chain = LLMChain(llm=self.openai_gpt4_llm, prompt=full_prompt_template)
|