wealthcoders commited on
Commit
ae154ba
·
verified ·
1 Parent(s): 68dd4db

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -8,7 +8,7 @@ import os
8
 
9
  class EndpointHandler:
10
  def __init__(self, model_dir = 'deepseek-ai/DeepSeek-OCR'):
11
- model_name = model_dir
12
 
13
  self.tokenizer = AutoTokenizer.from_pretrained(
14
  model_path,
 
8
 
9
  class EndpointHandler:
10
  def __init__(self, model_dir = 'deepseek-ai/DeepSeek-OCR'):
11
+ model_path = model_dir
12
 
13
  self.tokenizer = AutoTokenizer.from_pretrained(
14
  model_path,