k

app.py

@@ -14,7 +14,7 @@ print(f"device is :{device}")
 # init stable diffusion model
 pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4",  torch_dtype=th.float32, use_auth_token= HF_TOKEN).to(device)  #revision="fp16",
 
-def get_sd(translated_txt):
+def get_sd_old(translated_txt):
     scale=7.5 
     steps=45
     with autocast('cpu' if not th.cuda.is_available() else 'cuda'):
@@ -27,12 +27,12 @@ def get_sd(translated_txt):
 #headers = {"Authorization": f"Bearer {HF_TOKEN}"}
 sd_inf = gr.Blocks.load(name="spaces/stabilityai/stable-diffusion", use_auth_token=HF_TOKEN )#'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')
 
-def get_sd_old(translated_txt):
+def get_sd(translated_txt):
   print("******** Inside get_SD ********")
   print(f"translated_txt is : {translated_txt}")
   #sd_inf = gr.Blocks.load(name="spaces/stabilityai/stable-diffusion", use_auth_token='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')
   print(f"stable Diff inf is : {sd_inf}")
-  sd_img_gallery = sd_inf([translated_txt, 4,45, 7.5,1024], fn_index=1)[0]  #(prompt, samples, steps, scale, seed) #translated_txt
+  sd_img_gallery = sd_inf([translated_txt, 4,45, 7.5,1024], fn_index=2)[0]  #(prompt, samples, steps, scale, seed) #translated_txt
   
   return sd_img_gallery[0]