reichaves commited on
Commit
665116e
1 Parent(s): 94228b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -168,7 +168,8 @@ if groq_api_key and huggingface_api_token:
168
  os.environ["HUGGINGFACEHUB_API_TOKEN"] = huggingface_api_token
169
 
170
  # Inicializar o modelo de linguagem e embeddings
171
- llm = ChatGroq(groq_api_key=groq_api_key, model_name="Gemma2-9b-It", temperature=0)
 
172
  embeddings = HuggingFaceEmbeddings(model_name="all-MiniLM-L6-v2")
173
 
174
  session_id = st.text_input("Session ID", value="default_session")
 
168
  os.environ["HUGGINGFACEHUB_API_TOKEN"] = huggingface_api_token
169
 
170
  # Inicializar o modelo de linguagem e embeddings
171
+ #llm = ChatGroq(groq_api_key=groq_api_key, model_name="Gemma2-9b-It", temperature=0)
172
+ llm = ChatGroq(groq_api_key=groq_api_key, model_name="llama3-70b-8192, temperature=0)
173
  embeddings = HuggingFaceEmbeddings(model_name="all-MiniLM-L6-v2")
174
 
175
  session_id = st.text_input("Session ID", value="default_session")