asofter commited on
Commit
805526f
·
1 Parent(s): 1a0ab69

* aws update

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -164,7 +164,7 @@ def detect_azure(prompt: str) -> (bool, bool):
164
 
165
  def detect_aws_bedrock(prompt: str) -> (bool, bool):
166
  response = bedrock_runtime_client.apply_guardrail(
167
- guardrailIdentifier="arn:aws:bedrock:us-east-1:364432806369:guardrail/tx8t6psx14ho",
168
  guardrailVersion="1",
169
  source='INPUT',
170
  content=[
@@ -251,7 +251,7 @@ def execute(prompt: str) -> List[Union[str, bool, float]]:
251
  hf_api.upload_file(
252
  path_or_fileobj=fileobj,
253
  path_in_repo=result_path,
254
- repo_id="ProtectAI/prompt-injection-benchmark",
255
  repo_type="dataset",
256
  )
257
  logger.info(f"Stored prompt: {prompt}")
 
164
 
165
  def detect_aws_bedrock(prompt: str) -> (bool, bool):
166
  response = bedrock_runtime_client.apply_guardrail(
167
+ guardrailIdentifier="tx8t6psx14ho",
168
  guardrailVersion="1",
169
  source='INPUT',
170
  content=[
 
251
  hf_api.upload_file(
252
  path_or_fileobj=fileobj,
253
  path_in_repo=result_path,
254
+ repo_id="protectai/prompt-injection-benchmark",
255
  repo_type="dataset",
256
  )
257
  logger.info(f"Stored prompt: {prompt}")