shivi commited on
Commit
5a4920f
1 Parent(s): 1dccceb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -84,6 +84,8 @@ def replicate_api_inference(input_prompt):
84
  "aspect_ratio": "1:1",
85
  "output_format": "jpg",
86
  "output_quality": 80,
 
 
87
  "num_inference_steps": 4
88
  }
89
  image = replicate.run("black-forest-labs/flux-schnell",input=input_params)
 
84
  "aspect_ratio": "1:1",
85
  "output_format": "jpg",
86
  "output_quality": 80,
87
+ "enable_safety_checker": True,
88
+ "safety_tolerance": 1,
89
  "num_inference_steps": 4
90
  }
91
  image = replicate.run("black-forest-labs/flux-schnell",input=input_params)