Update handler.py

handler.py

@@ -19,7 +19,7 @@ class EndpointHandler():
         inputs = self.processor(image, return_tensors="pt").to(device)
 
         with torch.no_grad():
-            outputs = model(**inputs)
+            outputs = self.model(**inputs)
         
         last_hidden_states = outputs.last_hidden_state
         return last_hidden_states[2].tolist()