Update app.py
Browse files
app.py
CHANGED
@@ -21,7 +21,7 @@ pipe.enable_model_cpu_offload()
|
|
21 |
pipe.enable_vae_slicing()
|
22 |
|
23 |
def safety_check(user_prompt):
|
24 |
-
|
25 |
response = client.predict(
|
26 |
user_prompt, # str in 'User sent this' Textbox component
|
27 |
api_name="/infer"
|
|
|
21 |
pipe.enable_vae_slicing()
|
22 |
|
23 |
def safety_check(user_prompt):
|
24 |
+
|
25 |
response = client.predict(
|
26 |
user_prompt, # str in 'User sent this' Textbox component
|
27 |
api_name="/infer"
|