sberbank-ai commited on
Commit
1d9640c
1 Parent(s): 8b72d70

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -26,12 +26,13 @@ def get_config_and_download_weights(repo_id, device='cpu'):
26
  pipeline_config['main_process']['SegmPrediction']['config_path'] = segm_config_path
27
  pipeline_config['main_process']['SegmPrediction']['num_threads'] = 2
28
  pipeline_config['main_process']['SegmPrediction']['device'] = device
29
- pipeline_config['main_process']['SegmPrediction']['onnx'] = True
 
30
  pipeline_config['main_process']['OCRPrediction']['model_path'] = ocr_model_path
31
  pipeline_config['main_process']['OCRPrediction']['config_path'] = ocr_config_path
32
  pipeline_config['main_process']['OCRPrediction']['num_threads'] = 2
33
  pipeline_config['main_process']['OCRPrediction']['device'] = device
34
- pipeline_config['main_process']['OCRPrediction']['onnx'] = True
35
 
36
  # save pipeline_config
37
  with open(pipeline_config_path, 'w') as f:
 
26
  pipeline_config['main_process']['SegmPrediction']['config_path'] = segm_config_path
27
  pipeline_config['main_process']['SegmPrediction']['num_threads'] = 2
28
  pipeline_config['main_process']['SegmPrediction']['device'] = device
29
+ pipeline_config['main_process']['SegmPrediction']['runtime'] = "OpenVino"
30
+
31
  pipeline_config['main_process']['OCRPrediction']['model_path'] = ocr_model_path
32
  pipeline_config['main_process']['OCRPrediction']['config_path'] = ocr_config_path
33
  pipeline_config['main_process']['OCRPrediction']['num_threads'] = 2
34
  pipeline_config['main_process']['OCRPrediction']['device'] = device
35
+ pipeline_config['main_process']['OCRPrediction']['runtime'] = "OpenVino"
36
 
37
  # save pipeline_config
38
  with open(pipeline_config_path, 'w') as f: