Ishaan Gupta commited on
Commit
8009a47
1 Parent(s): 0361a01

updated to model path in handler

Browse files
Files changed (1) hide show
  1. handler.py +2 -2
handler.py CHANGED
@@ -10,8 +10,8 @@ class EndpointHandler():
10
  # Preload all the elements you are going to need at inference.
11
  # pseudo:
12
  # self.model= load_model(path)
13
- self.model = AutoModelForVision2Seq.from_pretrained("microsoft/kosmos-2-patch14-224").to("cuda")
14
- self.processor = AutoProcessor.from_pretrained("microsoft/kosmos-2-patch14-224")
15
 
16
  # prompt = "<grounding>An image of"
17
 
 
10
  # Preload all the elements you are going to need at inference.
11
  # pseudo:
12
  # self.model= load_model(path)
13
+ self.model = AutoModelForVision2Seq.from_pretrained(path).to("cuda")
14
+ self.processor = AutoProcessor.from_pretrained(path)
15
 
16
  # prompt = "<grounding>An image of"
17