SagayaAbinesh commited on
Commit
e3baf59
1 Parent(s): 1ce3a8f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -7,6 +7,7 @@ from langchain.chains import ConversationalRetrievalChain
7
  import time
8
  from langchain_google_genai import ChatGoogleGenerativeAI
9
  import streamlit as st
 
10
 
11
  st.set_page_config(page_title="GymGPT")
12
  col1, col2, col3 = st.columns([1,2,1])
@@ -79,10 +80,9 @@ ASSISTANT:
79
  prompt = PromptTemplate(template=prompt_template,
80
  input_variables=['question', 'context', 'chat_history'])
81
 
82
- import os
83
- os.environ['gemini_api']
84
 
85
- llm = ChatGoogleGenerativeAI(model="gemini-1.5-flash")
86
 
87
  # Create a conversational chain using only your database retriever
88
  qa = ConversationalRetrievalChain.from_llm(
 
7
  import time
8
  from langchain_google_genai import ChatGoogleGenerativeAI
9
  import streamlit as st
10
+ import os
11
 
12
  st.set_page_config(page_title="GymGPT")
13
  col1, col2, col3 = st.columns([1,2,1])
 
80
  prompt = PromptTemplate(template=prompt_template,
81
  input_variables=['question', 'context', 'chat_history'])
82
 
83
+ # GOOGLE_API_KEY = os.environ['gemini_api']
 
84
 
85
+ llm = ChatGoogleGenerativeAI(model="gemini-1.5-flash", google_api_key=os.environ['gemini_api'])
86
 
87
  # Create a conversational chain using only your database retriever
88
  qa = ConversationalRetrievalChain.from_llm(