Iliassti commited on
Commit
5cc47df
·
1 Parent(s): e1c6438

ChatOpenAI

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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(model_name="gpt-4", temperature=0, max_tokens=256)
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)