Spaces:

MJobe
/

document-vqa-v2

Sleeping

MJobe commited on Dec 12, 2023

Commit

41ddd5c

1 Parent(s): 41d335c

Update main.py

Files changed (1) hide show

main.py CHANGED Viewed

@@ -40,10 +40,10 @@ async def perform_document_qa(
                 question.strip()
             )
-            # Print the result for debugging
-            print(result)
-            answers_dict[question] = result['answer']
         return answers_dict
     except Exception as e:
@@ -69,10 +69,10 @@ async def load_file(
                 question.strip()
             )
-            # Print the result for debugging
-            print(result)
-            answers_dict[question] = result['answer']
         return answers_dict
     except Exception as e:

                 question.strip()
             )
+            # Access the 'answer' key from the first item in the result list
+            answer = result[0]['answer']
+            answers_dict[question] = answer
         return answers_dict
     except Exception as e:
                 question.strip()
             )
+            # Access the 'answer' key from the first item in the result list
+            answer = result[0]['answer']
+            answers_dict[question] = answer
         return answers_dict
     except Exception as e: