jchen8000 commited on
Commit
7279c69
·
verified ·
1 Parent(s): 055baa9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import random
2
  import gradio as gr
3
  from langchain_community.document_loaders import PyPDFLoader
@@ -70,7 +71,7 @@ def generate_response(query, history, model, temperature, max_tokens, top_p, see
70
  if seed == 0:
71
  seed = random.randint(1, 100000)
72
 
73
- llm = ChatGroq(groq_api_key=userdata.get('GROQ_API_KEY'), model=model)
74
 
75
  custom_rag_prompt = PromptTemplate.from_template(template)
76
 
 
1
+ import os
2
  import random
3
  import gradio as gr
4
  from langchain_community.document_loaders import PyPDFLoader
 
71
  if seed == 0:
72
  seed = random.randint(1, 100000)
73
 
74
+ llm = ChatGroq(groq_api_key=os.environ.get("GROQ_API_KEY"), model=model)
75
 
76
  custom_rag_prompt = PromptTemplate.from_template(template)
77