zliang commited on
Commit
e2f8798
·
verified ·
1 Parent(s): 6567523

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -26,6 +26,8 @@ model = YOLO("best.pt")
26
  openai_api_key = os.environ.get("openai_api_key")
27
  MAX_FILE_SIZE = 50 * 1024 * 1024 # 50MB
28
 
 
 
29
  # Utility functions
30
  @st.cache_data(show_spinner=False, ttl=3600)
31
  def clean_text(text):
@@ -181,7 +183,7 @@ def add_interactive_citations(summary_text, source_chunks):
181
  @handle_errors
182
  def qa_pdf(_pdf_file_path, query, num_clusters=5):
183
  embeddings_model = OpenAIEmbeddings(model="text-embedding-3-small", api_key=openai_api_key)
184
- llm = ChatOpenAI(model="gpt-4", api_key=openai_api_key, temperature=0.3)
185
 
186
  # Load PDF with page numbers
187
  loader = PyMuPDFLoader(_pdf_file_path)
 
26
  openai_api_key = os.environ.get("openai_api_key")
27
  MAX_FILE_SIZE = 50 * 1024 * 1024 # 50MB
28
 
29
+ llm = ChatOpenAI(model="gpt-3.5-turbo", api_key=openai_api_key, temperature=0.3)
30
+
31
  # Utility functions
32
  @st.cache_data(show_spinner=False, ttl=3600)
33
  def clean_text(text):
 
183
  @handle_errors
184
  def qa_pdf(_pdf_file_path, query, num_clusters=5):
185
  embeddings_model = OpenAIEmbeddings(model="text-embedding-3-small", api_key=openai_api_key)
186
+ # llm = ChatOpenAI(model="gpt-3.5-turbo", api_key=openai_api_key, temperature=0.3)
187
 
188
  # Load PDF with page numbers
189
  loader = PyMuPDFLoader(_pdf_file_path)