anal

app.py

@@ -21,13 +21,13 @@ prompt=st.text_input('Enter Your Prompt')
 
 model_id = "CompVis/stable-diffusion-v1-4"
 device = "cpu"
-
+st.info('1')
 #___________________________________________________________________________________________________________________________
 
 pipe = StableDiffusionPipeline.from_pretrained(model_id, use_auth_token='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', torch_dtype=torch.float32)
 def dummy(images, **kwargs): return images, False 
 pipe.safety_checker = dummy
-
+st.info(2)
 #___________________________________________________________________________________________________________________________
 
 def infer(prompt, width, height, steps, scale, seed):      
@@ -48,14 +48,15 @@ def infer(prompt, width, height, steps, scale, seed):
             guidance_scale=scale)
 
     return images_list["sample"]
-
+    st.info(5)
+st.info(3)
 #___________________________________________________________________________________________________________________________
 
 def onclick(prompt):
     st.image(infer(prompt,512,512,30,7.5,-1))
-
 if prompt==True:
     onclick(prompt)
+    st.info(4)
 
 #___________________________________________________________________________________________________________________________