Spaces:
Runtime error
Runtime error
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
|