Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -81,6 +81,7 @@ def initialize_llmchain(llm_model, temperature, max_tokens, top_k, vector_db, pr
|
|
| 81 |
return qa_chain
|
| 82 |
|
| 83 |
# Initialize database
|
|
|
|
| 84 |
def initialize_database(list_file_obj, progress=gr.Progress()):
|
| 85 |
# Create a list of documents (when valid)
|
| 86 |
list_file_path = [x.name for x in list_file_obj if x is not None]
|
|
|
|
| 81 |
return qa_chain
|
| 82 |
|
| 83 |
# Initialize database
|
| 84 |
+
@spaces.GPU(duration=60)
|
| 85 |
def initialize_database(list_file_obj, progress=gr.Progress()):
|
| 86 |
# Create a list of documents (when valid)
|
| 87 |
list_file_path = [x.name for x in list_file_obj if x is not None]
|