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)
|