shivanis14 commited on
Commit
6233dee
·
verified ·
1 Parent(s): 43f54a5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -20,12 +20,12 @@ print(f"api key is {api_key}")
20
  @cl.on_chat_start
21
  async def start_llm():
22
  print("Initializing llm...")
23
- #llm = ChatOpenAI(model='gpt-4o-mini',
24
- # temperature=0.5, api_key = api_key)
25
- model_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
26
- llm = HuggingFaceEndpoint(
27
- repo_id=model_id, max_length=2000, temperature=0.5, token=os.getenv('HF_READ_TOKEN')
28
- )
29
  print("llm initialized!")
30
  conversation_memory = ConversationBufferMemory(memory_key="chat_history",
31
  max_len=50,
 
20
  @cl.on_chat_start
21
  async def start_llm():
22
  print("Initializing llm...")
23
+ llm = ChatOpenAI(model='gpt-4o-mini',
24
+ temperature=0.5, api_key = api_key)
25
+ #model_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
26
+ #llm = HuggingFaceEndpoint(
27
+ # repo_id=model_id, max_length=2000, temperature=0.5, token=os.getenv('HF_READ_TOKEN')
28
+ #)
29
  print("llm initialized!")
30
  conversation_memory = ConversationBufferMemory(memory_key="chat_history",
31
  max_len=50,