Update app.py
Browse files
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=
|
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 |
|