Spaces:
Runtime error
Runtime error
Commit
·
14fefb4
1
Parent(s):
6e8fb2b
Update app.py
Browse files
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(
|