kavg commited on
Commit
613ad82
·
1 Parent(s): 7b11b8d

Removed vision_client from config dictionary

Browse files
Files changed (1) hide show
  1. main.py +3 -2
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
- config['vision_client'] = ocr.VisionClient(settings.GCV_AUTH)
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
- ocr_df = config['vision_client'].ocr(content, image)
 
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