omkar56 commited on
Commit
7547c3f
·
1 Parent(s): bf8046c

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -5
main.py CHANGED
@@ -1,4 +1,5 @@
1
  import os
 
2
  from fastapi import FastAPI, File, Request, UploadFile, Body, Depends, HTTPException
3
  from fastapi.security.api_key import APIKeyHeader
4
  from typing import Optional, Annotated
@@ -24,17 +25,12 @@ async def ocr(
24
  api_key: str = Depends(get_api_key),
25
  image: UploadFile = File(...),
26
  # languages: list = Body(["eng"])
27
- # request: Request,
28
- # body: dict = Body(...),
29
  ):
30
- # print("[where?] outside try block")
31
- # print("[image]", image)
32
  try:
33
  content = await image.read()
34
  image = Image.open(BytesIO(content))
35
  # text = pytesseract.image_to_string(image, lang="+".join(languages))
36
  text = pytesseract.image_to_string(image, lang = 'eng')
37
- # message = "file uploaded"
38
  except Exception as e:
39
  return {"error": str(e)}, 500
40
 
 
1
  import os
2
+ os.system("sudo apt-get install tesseract-ocr")
3
  from fastapi import FastAPI, File, Request, UploadFile, Body, Depends, HTTPException
4
  from fastapi.security.api_key import APIKeyHeader
5
  from typing import Optional, Annotated
 
25
  api_key: str = Depends(get_api_key),
26
  image: UploadFile = File(...),
27
  # languages: list = Body(["eng"])
 
 
28
  ):
 
 
29
  try:
30
  content = await image.read()
31
  image = Image.open(BytesIO(content))
32
  # text = pytesseract.image_to_string(image, lang="+".join(languages))
33
  text = pytesseract.image_to_string(image, lang = 'eng')
 
34
  except Exception as e:
35
  return {"error": str(e)}, 500
36