joshuadunlop commited on
Commit
665f9ef
·
1 Parent(s): 7a3a8ac

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -98,10 +98,10 @@ def load_recommender(path, start_page=1):
98
  return 'Corpus Loaded.'
99
 
100
  def generate_text(openAI_key,prompt, engine="text-davinci-003"):
101
- try:
102
- openai.api_key = openAI_key
103
- except Exception as e:
104
- raise ValueError("Failed to set OpenAI API key: " + str(e))
105
  completions = openai.Completion.create(
106
  engine=engine,
107
  prompt=prompt,
@@ -191,10 +191,10 @@ for i in range(row_count):
191
  st.error('Question field is empty')
192
  else:
193
  glob_url = url
194
- try:
195
- download_pdf(glob_url, 'corpus.pdf')
196
- except Exception as e:
197
- raise ValueError("Failed to download and process the PDF: " + str(e))
198
  load_recommender('corpus.pdf')
199
 
200
  answer = generate_answer(question,openAI_key)
 
98
  return 'Corpus Loaded.'
99
 
100
  def generate_text(openAI_key,prompt, engine="text-davinci-003"):
101
+ try:
102
+ openai.api_key = openAI_key
103
+ except Exception as e:
104
+ raise ValueError("Failed to set OpenAI API key: " + str(e))
105
  completions = openai.Completion.create(
106
  engine=engine,
107
  prompt=prompt,
 
191
  st.error('Question field is empty')
192
  else:
193
  glob_url = url
194
+ try:
195
+ download_pdf(glob_url, 'corpus.pdf')
196
+ except Exception as e:
197
+ raise ValueError("Failed to download and process the PDF: " + str(e))
198
  load_recommender('corpus.pdf')
199
 
200
  answer = generate_answer(question,openAI_key)