adasdimchom
commited on
Commit
·
016cd63
1
Parent(s):
f69e8d5
Upload handler.py
Browse files- 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
|