bhaskartripathi commited on
Commit
14fefb4
·
1 Parent(s): 6e8fb2b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -110,6 +110,11 @@ def load_recommender(path, start_page=1, end_page=None):
110
  recommender.fitted = True
111
  return "Embeddings loaded from file"
112
 
 
 
 
 
 
113
  texts = pdf_to_text(path, start_page=start_page, end_page=end_page)
114
  chunks = text_to_chunks(texts, start_page=start_page)
115
  recommender.fit(chunks)
@@ -118,6 +123,7 @@ def load_recommender(path, start_page=1, end_page=None):
118
 
119
 
120
 
 
121
  def generate_text(openAI_key,prompt, engine="text-davinci-003"):
122
  openai.api_key = openAI_key
123
  completions = openai.Completion.create(
 
110
  recommender.fitted = True
111
  return "Embeddings loaded from file"
112
 
113
+ if start_page:
114
+ start_page = int(start_page)
115
+ if end_page:
116
+ end_page = int(end_page)
117
+
118
  texts = pdf_to_text(path, start_page=start_page, end_page=end_page)
119
  chunks = text_to_chunks(texts, start_page=start_page)
120
  recommender.fit(chunks)
 
123
 
124
 
125
 
126
+
127
  def generate_text(openAI_key,prompt, engine="text-davinci-003"):
128
  openai.api_key = openAI_key
129
  completions = openai.Completion.create(