TanU21 commited on
Commit
d0f318a
·
verified ·
1 Parent(s): a037aba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -3,9 +3,14 @@ from langchain_core.prompts import ChatPromptTemplate
3
  import gradio as gr
4
  import os
5
 
 
6
 
7
- llm = ChatOllama(model="llama3.2", base_url = "https://alpaca-upright-vulture.ngrok-free.app", temperature=0.5)
8
-
 
 
 
 
9
  prompt = ChatPromptTemplate.from_messages(
10
  [
11
  ("system","Generate a list of synonyms for the following word, only give 5 words with examples."),
 
3
  import gradio as gr
4
  import os
5
 
6
+ groq_api_key = os.environ.get("GROQ_API_KEY")
7
 
8
+ llm = ChatGroq(
9
+ model="mixtral-8x7b-32768",
10
+ temperature=0.7,
11
+ api_key=groq_api_key,
12
+ verbose = False
13
+ )
14
  prompt = ChatPromptTemplate.from_messages(
15
  [
16
  ("system","Generate a list of synonyms for the following word, only give 5 words with examples."),