wealthcoders commited on
Commit
e630943
·
verified ·
1 Parent(s): 69a70df

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -12,7 +12,7 @@ class EndpointHandler:
12
  model_path = model_dir
13
 
14
  self.model = AutoModelForImageTextToText.from_pretrained(model_path, dtype="auto", device_map="auto")
15
- selfprocessor = AutoProcessor.from_pretrained(model_path)
16
 
17
  def __call__(self, data: Dict[str, Any]) -> str:
18
  try:
 
12
  model_path = model_dir
13
 
14
  self.model = AutoModelForImageTextToText.from_pretrained(model_path, dtype="auto", device_map="auto")
15
+ self.processor = AutoProcessor.from_pretrained(model_path)
16
 
17
  def __call__(self, data: Dict[str, Any]) -> str:
18
  try: