fffiloni commited on
Commit
44dd015
·
verified ·
1 Parent(s): 17194a9

api update

Browse files
Files changed (1) hide show
  1. app.py +7 -4
app.py CHANGED
@@ -10,12 +10,15 @@ from gradio_client import Client
10
 
11
 
12
  def safety_check(user_prompt):
13
- client = Client("fffiloni/safety-checker-bot", hf_token=hf_token)
 
14
  response = client.predict(
15
- "consistent-character space", # str source space
16
- user_prompt, # str in 'User sent this' Textbox component
17
- api_name="/infer"
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