Eeman Majumder commited on
Commit
5dd9c5c
·
1 Parent(s): 1b54286
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -27,7 +27,7 @@ st.info('1')
27
  pipe = StableDiffusionPipeline.from_pretrained(model_id, use_auth_token='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', torch_dtype=torch.float32)
28
  def dummy(images, **kwargs): return images, False
29
  pipe.safety_checker = dummy
30
- st.info(2)
31
  #___________________________________________________________________________________________________________________________
32
 
33
  def infer(prompt, width, height, steps, scale, seed):
@@ -48,15 +48,15 @@ def infer(prompt, width, height, steps, scale, seed):
48
  guidance_scale=scale)
49
 
50
  return images_list["sample"]
51
- st.info(5)
52
- st.info(3)
53
  #___________________________________________________________________________________________________________________________
54
 
55
  def onclick(prompt):
56
  st.image(infer(prompt,512,512,30,7.5,-1))
57
  if prompt==True:
58
  onclick(prompt)
59
- st.info(4)
60
 
61
  #___________________________________________________________________________________________________________________________
62
 
 
27
  pipe = StableDiffusionPipeline.from_pretrained(model_id, use_auth_token='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', torch_dtype=torch.float32)
28
  def dummy(images, **kwargs): return images, False
29
  pipe.safety_checker = dummy
30
+ st.info('2')
31
  #___________________________________________________________________________________________________________________________
32
 
33
  def infer(prompt, width, height, steps, scale, seed):
 
48
  guidance_scale=scale)
49
 
50
  return images_list["sample"]
51
+ st.info('5')
52
+ st.info('3')
53
  #___________________________________________________________________________________________________________________________
54
 
55
  def onclick(prompt):
56
  st.image(infer(prompt,512,512,30,7.5,-1))
57
  if prompt==True:
58
  onclick(prompt)
59
+ st.info('4')
60
 
61
  #___________________________________________________________________________________________________________________________
62