Update app.py
Browse files
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-
|
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)
|