Removed vision_client from config dictionary
Browse files
main.py
CHANGED
@@ -18,7 +18,7 @@ async def lifespan(app: FastAPI):
|
|
18 |
settings = Settings()
|
19 |
config['settings'] = settings
|
20 |
config['device'] = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
21 |
-
|
22 |
config['processor'] = Preprocessor(settings.TOKENIZER)
|
23 |
config['tokenizer'] = AutoTokenizer.from_pretrained(settings.TOKENIZER)
|
24 |
config['ser_model'] = LiltForTokenClassification.from_pretrained(settings.SER_MODEL)
|
@@ -70,7 +70,8 @@ def ApplyOCR(content):
|
|
70 |
except:
|
71 |
raise HTTPException(status_code=400, detail="Invalid image")
|
72 |
try:
|
73 |
-
|
|
|
74 |
except:
|
75 |
raise HTTPException(status_code=400, detail="OCR process failed")
|
76 |
return ocr_df, image
|
|
|
18 |
settings = Settings()
|
19 |
config['settings'] = settings
|
20 |
config['device'] = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
21 |
+
|
22 |
config['processor'] = Preprocessor(settings.TOKENIZER)
|
23 |
config['tokenizer'] = AutoTokenizer.from_pretrained(settings.TOKENIZER)
|
24 |
config['ser_model'] = LiltForTokenClassification.from_pretrained(settings.SER_MODEL)
|
|
|
70 |
except:
|
71 |
raise HTTPException(status_code=400, detail="Invalid image")
|
72 |
try:
|
73 |
+
vision_client = ocr.VisionClient(config['settings'].GCV_AUTH)
|
74 |
+
ocr_df = vision_client.ocr(content, image)
|
75 |
except:
|
76 |
raise HTTPException(status_code=400, detail="OCR process failed")
|
77 |
return ocr_df, image
|