yuvaranianandhan24 commited on
Commit
2c0d74d
Β·
verified Β·
1 Parent(s): c93c025

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -8,6 +8,7 @@ from langchain.llm import CTransformers
8
  from langchain.chains import LLMChain
9
  from langchain.prompts import PromptTemplate
10
  from PIL import Image
 
11
 
12
  def main():
13
  st.title("Invoice Entity Extractor πŸ“š")
@@ -20,6 +21,8 @@ def main():
20
  elif uploaded_image is not None:
21
  process_image(uploaded_image)
22
 
 
 
23
  def process_pdf(uploaded_file):
24
  # Process the uploaded PDF file
25
  with NamedTemporaryFile(delete=False) as temp_file:
 
8
  from langchain.chains import LLMChain
9
  from langchain.prompts import PromptTemplate
10
  from PIL import Image
11
+ import os
12
 
13
  def main():
14
  st.title("Invoice Entity Extractor πŸ“š")
 
21
  elif uploaded_image is not None:
22
  process_image(uploaded_image)
23
 
24
+ api_token = os.getenv('HF_TOKEN')
25
+
26
  def process_pdf(uploaded_file):
27
  # Process the uploaded PDF file
28
  with NamedTemporaryFile(delete=False) as temp_file: