Spaces:
Running
on
L40S
Running
on
L40S
api update
Browse files
app.py
CHANGED
@@ -10,12 +10,15 @@ from gradio_client import Client
|
|
10 |
|
11 |
|
12 |
def safety_check(user_prompt):
|
13 |
-
|
|
|
14 |
response = client.predict(
|
15 |
-
|
16 |
-
|
17 |
-
|
18 |
)
|
|
|
|
|
19 |
return response
|
20 |
|
21 |
from utils.gradio_helpers import parse_outputs, process_outputs
|
|
|
10 |
|
11 |
|
12 |
def safety_check(user_prompt):
|
13 |
+
|
14 |
+
client = Client("fffiloni/safety-checker-bot")
|
15 |
response = client.predict(
|
16 |
+
source_space="consistent-character space",
|
17 |
+
user_prompt=user_prompt,
|
18 |
+
api_name="/infer"
|
19 |
)
|
20 |
+
print(response)
|
21 |
+
|
22 |
return response
|
23 |
|
24 |
from utils.gradio_helpers import parse_outputs, process_outputs
|