adasdimchom commited on
Commit
016cd63
·
1 Parent(s): f69e8d5

Upload handler.py

Browse files
Files changed (1) hide show
  1. handler.py +0 -4
handler.py CHANGED
@@ -37,8 +37,4 @@ class EndpointHandler():
37
  output = self.model.generate(**processed_image)
38
  text_output = self.processor.decode(output[0], skip_special_tokens=True)
39
  result["text_output"] = text_output
40
- #feature_vector = output
41
- #shape = output.shape
42
- #result["feature_vector"] = feature_vector
43
- #result["shape"] = shape
44
  return result
 
37
  output = self.model.generate(**processed_image)
38
  text_output = self.processor.decode(output[0], skip_special_tokens=True)
39
  result["text_output"] = text_output
 
 
 
 
40
  return result