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
|