Chan-Y commited on
Commit
fd85229
1 Parent(s): ac65db8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -49,7 +49,7 @@ def process_file(file):
49
 
50
  return [Document(text=text)]
51
 
52
- def handle_file_upload(file, llm_name):
53
  global query_engine
54
  if torch.cuda.is_available():
55
  torch.cuda.empty_cache()
@@ -72,7 +72,7 @@ def handle_file_upload(file, llm_name):
72
  return response
73
 
74
  def document_qa(file_upload, llm_choice, question_input):
75
- response = handle_file_upload(file_upload, llm_choice)
76
  return response
77
 
78
 
 
49
 
50
  return [Document(text=text)]
51
 
52
+ def handle_file_upload(file, llm_name, question):
53
  global query_engine
54
  if torch.cuda.is_available():
55
  torch.cuda.empty_cache()
 
72
  return response
73
 
74
  def document_qa(file_upload, llm_choice, question_input):
75
+ response = handle_file_upload(file_upload, llm_choice, question_input)
76
  return response
77
 
78