* aws update
Browse files
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="
|
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="
|
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}")
|