scholarly360 commited on
Commit
238e05a
·
verified ·
1 Parent(s): 88fb826

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -6,6 +6,7 @@ import os
6
  import chromadb
7
  import uuid
8
  from pathlib import Path
 
9
 
10
  os.environ['OPENAI_API_KEY'] = os.environ['OPEN_API_KEY']
11
  st.title("Contracts Summary ")
@@ -51,6 +52,7 @@ documents = []
51
 
52
 
53
  def get_summary_single_doc(text):
 
54
  from langchain.llms import OpenAI
55
  from langchain.chains.summarize import load_summarize_chain
56
  from langchain.text_splitter import CharacterTextSplitter
@@ -86,7 +88,7 @@ def get_summary_single_doc(text):
86
  #Define the LLM
87
  # here we are using OpenAI's ChatGPT
88
  from langchain.chat_models import ChatOpenAI
89
- model_name = "gpt-3.5-turbo"
90
  llm=ChatOpenAI(temperature=0, openai_api_key=LLM_KEY, model_name=model_name)
91
 
92
  refine_chain = load_summarize_chain(
 
6
  import chromadb
7
  import uuid
8
  from pathlib import Path
9
+ import time
10
 
11
  os.environ['OPENAI_API_KEY'] = os.environ['OPEN_API_KEY']
12
  st.title("Contracts Summary ")
 
52
 
53
 
54
  def get_summary_single_doc(text):
55
+ time.sleep(5)
56
  from langchain.llms import OpenAI
57
  from langchain.chains.summarize import load_summarize_chain
58
  from langchain.text_splitter import CharacterTextSplitter
 
88
  #Define the LLM
89
  # here we are using OpenAI's ChatGPT
90
  from langchain.chat_models import ChatOpenAI
91
+ model_name = "gpt-4o"
92
  llm=ChatOpenAI(temperature=0, openai_api_key=LLM_KEY, model_name=model_name)
93
 
94
  refine_chain = load_summarize_chain(