srijaydeshpande commited on
Commit
4c9d420
·
verified ·
1 Parent(s): b355172

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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]