Update app.py

app.py

@@ -78,7 +78,7 @@ def get_page_result(raw_result, question):
             "question": question,
             "answer": raw_result[0]['answer']
         },
-        confidence: raw_result[0]['score']
+        "confidence": raw_result[0]['score']
     }
     return page_result
     
@@ -103,7 +103,7 @@ def qna_pdf_content(image_paths, question, vision_qna, short_circuit_threshold):
 
 @spaces.GPU
 def answer_the_question_for_doc(file_input, question, short_circuit_threshold):
-  vision_qna = pipeline("document-question-answering", model="impira/layoutlm-document-qa")#, device=0)
+  vision_qna = pipeline("document-question-answering", model="impira/layoutlm-document-qa", device=0)
  
   if file_input is not None and question is not None:
     print(f"File type: {type(file_input)}")