Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import random
 import time
 
+from langchain.llms import OpenAI, OpenAIChat
 from langchain.chat_models import ChatOpenAI
 from langchain.embeddings import HuggingFaceEmbeddings
 from langchain.vectorstores import Pinecone
@@ -47,13 +48,19 @@ Please insert your question and click 'Submit'
 
 def init_model(openai_key):
     try:
+        
+        os.environ["OPENAI_API_KEY"] = openai_key
+        
         embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2")
 
         pinecone.init(api_key     = PINECONE_KEY,
                       environment = PINECONE_ENV)
 
-        llm = ChatOpenAI(temperature    = OPENAI_TEMP,
-                         openai_api_key = openai_key)
+        llm = OpenAI(temperature=OPENAI_TEMP,
+                     model_name="gpt-3.5-turbo-0301")
+        
+        # ChatOpenAI(temperature    = OPENAI_TEMP, openai_api_key = openai_key)
+        
 
         global db
         db = Pinecone.from_existing_index(index_name = PINECONE_INDEX,