Upload handler.py

handler.py

@@ -37,8 +37,8 @@ class EndpointHandler():
         output = self.model.generate(**processed_image)
         text_output = self.processor.decode(output[0], skip_special_tokens=True)
         result["text_output"] = text_output
-        feature_vector = output
-        shape = output.shape
+        #feature_vector = output
+        #shape = output.shape
         #result["feature_vector"] = feature_vector
-        result["shape"] = shape
+        #result["shape"] = shape
         return result