Update app.py

app.py

@@ -8,13 +8,22 @@ processor = LayoutLMv3Processor.from_pretrained("microsoft/layoutlmv3-base")
 model = LayoutLMv3ForQuestionAnswering.from_pretrained("microsoft/layoutlmv3-base")
 
 def process_question(question, document):
+    print(f"Debug - Processing Question: {question}")
+    
     encoding = processor(document, question, return_tensors="pt")
+    print(f"Debug - Encoding Input IDs: {encoding.input_ids}")
 
     outputs = model(**encoding)
+    print(f"Debug - Model Outputs: {outputs}") 
 
     predicted_start_idx = outputs.start_logits.argmax(-1).item()
     predicted_end_idx = outputs.end_logits.argmax(-1).item()
 
+    # Check if indices are valid
+    if predicted_start_idx < 0 or predicted_end_idx < 0:
+        print(f"Warning - Invalid prediction indices: start={predicted_start_idx}, end={predicted_end_idx}")
+        return ""
+
     answer_tokens = encoding.input_ids.squeeze()[predicted_start_idx: predicted_end_idx + 1]
     answer = processor.tokenizer.decode(answer_tokens)