vasilee commited on
Commit
5714343
·
verified ·
1 Parent(s): 780929c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ os.system('python download_models_hf.py')
21
  with open('/home/user/magic-pdf.json', 'r') as file:
22
  data = json.load(file)
23
 
24
- data['device-mode'] = "cuda"
25
  if os.getenv('apikey'):
26
  data['llm-aided-config']['title_aided']['api_key'] = os.getenv('apikey')
27
  data['llm-aided-config']['title_aided']['enable'] = True
 
21
  with open('/home/user/magic-pdf.json', 'r') as file:
22
  data = json.load(file)
23
 
24
+ data['device-mode'] = "cpu"
25
  if os.getenv('apikey'):
26
  data['llm-aided-config']['title_aided']['api_key'] = os.getenv('apikey')
27
  data['llm-aided-config']['title_aided']['enable'] = True