arifa2399 commited on
Commit
e955841
·
verified ·
1 Parent(s): 12cd783

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import streamlit as st
2
- from langchain_openai import ChatOpenAI
3
  from langchain_core.prompts import ChatPromptTemplate
 
4
  import os
5
 
6
  st.title("Translation Using OpenAI")
@@ -21,7 +21,7 @@ output_language = st.selectbox("Select your output language", languages)
21
  os.environ['GOOGLE_API_KEY'] = os.getenv("geminiapi_key")
22
 
23
  # Initialize the OpenAI language model
24
- llm = ChatOpenAI(model="gpt-3.5-turbo", temperature=0, max_tokens=80)
25
 
26
  # Create the prompt template
27
  prompt = ChatPromptTemplate.from_messages([
 
1
  import streamlit as st
 
2
  from langchain_core.prompts import ChatPromptTemplate
3
+ from langchain_google_genai import ChatGoogleGenerativeAI
4
  import os
5
 
6
  st.title("Translation Using OpenAI")
 
21
  os.environ['GOOGLE_API_KEY'] = os.getenv("geminiapi_key")
22
 
23
  # Initialize the OpenAI language model
24
+ llm = ChatGoogleGenerativeAI(model="gemini-pro", temperature=0, max_tokens=80)
25
 
26
  # Create the prompt template
27
  prompt = ChatPromptTemplate.from_messages([