Update app.py

reichaves · Sep 30, 2024 · 665116e · 665116e
1 parent 94228b5
commit 665116e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/app.py b/app.py
@@ -168,7 +168,8 @@
     os.environ["HUGGINGFACEHUB_API_TOKEN"] = huggingface_api_token
 
     # Inicializar o modelo de linguagem e embeddings
-    llm = ChatGroq(groq_api_key=groq_api_key, model_name="Gemma2-9b-It", temperature=0)
+    #llm = ChatGroq(groq_api_key=groq_api_key, model_name="Gemma2-9b-It", temperature=0)
+    llm = ChatGroq(groq_api_key=groq_api_key, model_name="llama3-70b-8192, temperature=0)
     embeddings = HuggingFaceEmbeddings(model_name="all-MiniLM-L6-v2")
 
     session_id = st.text_input("Session ID", value="default_session")