Update app.py

app.py

@@ -5,9 +5,9 @@ from transformers import pipeline
 
 
 
-st.title("Question - Answering AI App")
+st.title("Docs Answering AI App")
 st.write("---")
-st.write("### Give Content to the AI and quiz it based on the content.")
+st.write("### Give Doc Receipt Images to the AI and ask it whats on the receipt.")
 st.warning("**Powered by AI Language Models...**")
 
 st.write("---")
@@ -32,4 +32,25 @@ if uploaded_file is not None:
 
     # Can be used wherever a "file-like" object is accepted:
     dataframe = pd.read_csv(uploaded_file)
-    st.write(dataframe)
+    st.write(dataframe)
+
+
+nlp = pipeline(
+    "document-question-answering",
+    model="impira/layoutlm-document-qa",
+)
+
+res = nlp(
+    "https://templates.invoicehome.com/invoice-template-us-neat-750px.png",
+    "What is the invoice number?"
+)
+
+st.write(res['answer'])
+# {'score': 0.9943977, 'answer': 'us-001', 'start': 15, 'end': 15}
+
+res = nlp(
+    "https://miro.medium.com/max/787/1*iECQRIiOGTmEFLdWkVIH2g.jpeg",
+    "What is the purchase amount?"
+)
+
+st.write(res['answer'])