linnthit111 commited on
Commit
c564cad
·
verified ·
1 Parent(s): f77a1f0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -20,6 +20,7 @@ pipe = pipe.to(device)
20
  MAX_SEED = np.iinfo(np.int32).max
21
  MAX_IMAGE_SIZE = 300
22
 
 
23
 
24
  # @spaces.GPU #[uncomment to use ZeroGPU]
25
  def infer(
@@ -39,6 +40,7 @@ def infer(
39
  generator = torch.Generator().manual_seed(seed)
40
 
41
  image = pipe(
 
42
  prompt=prompt,
43
  negative_prompt=negative_prompt,
44
  guidance_scale=guidance_scale,
 
20
  MAX_SEED = np.iinfo(np.int32).max
21
  MAX_IMAGE_SIZE = 300
22
 
23
+ # ပုံဖန်တီးတဲ့အခါ safety_checker ကို
24
 
25
  # @spaces.GPU #[uncomment to use ZeroGPU]
26
  def infer(
 
40
  generator = torch.Generator().manual_seed(seed)
41
 
42
  image = pipe(
43
+ safety_checker=None
44
  prompt=prompt,
45
  negative_prompt=negative_prompt,
46
  guidance_scale=guidance_scale,