Update handler.py

handler.py

@@ -270,23 +270,23 @@ class EndpointHandler:
         decoded_output = self.tokenizer.decode(output[0], skip_special_tokens=True)
         
         # Extract the generated part (after the prompt)
-        response_text = decoded_output[len(prompt):].strip()
-        print(response_text)
+        # response_text = decoded_output[len(prompt):].strip()
+        # print(response_text)
         
-        # Parse the response to extract safety assessment
-        safety_result = self.extract_and_parse_json(response_text)
+        # # Parse the response to extract safety assessment
+        # safety_result = self.extract_and_parse_json(response_text)
         
-        # Determine if the input is safe or not
-        is_safe = safety_result.get("Prompt Safety", "").lower() == "safe" and \
-                 safety_result.get("Response Safety", "").lower() == "safe"
+        # # Determine if the input is safe or not
+        # is_safe = safety_result.get("Prompt Safety", "").lower() == "safe" and \
+        #          safety_result.get("Response Safety", "").lower() == "safe"
         
-        # Prepare the final response
-        response = {
-            "is_safe": is_safe,
-            "safety_result": safety_result
-        }
+        # # Prepare the final response
+        # response = {
+        #     "is_safe": is_safe,
+        #     "safety_result": safety_result
+        # }
         
-        return response
+        return decoded_output 
 
 # For local testing
 if __name__ == "__main__":