LDanielBlueway commited on
Commit
c4dbae8
·
verified ·
1 Parent(s): 6661511

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +2 -0
handler.py CHANGED
@@ -4,6 +4,8 @@ import base64
4
  import logging
5
  from PIL import Image
6
  import numpy as np
 
 
7
 
8
  class EndpointHandler():
9
  def __init__(self, path=""):
 
4
  import logging
5
  from PIL import Image
6
  import numpy as np
7
+ from transformers import AutoModel
8
+
9
 
10
  class EndpointHandler():
11
  def __init__(self, path=""):