l

app.py

@@ -12,7 +12,7 @@ device = th.device('cpu' if not th.cuda.is_available() else 'cuda')
 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",
+#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_old(translated_txt):
     scale=7.5 
@@ -32,7 +32,7 @@ def get_sd(translated_txt):
   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, float(4),float(45), 7.5,float(1024), fn_index=2)[0]  #(prompt, samples, steps, scale, seed) #translated_txt
+  sd_img_gallery = sd_inf(translated_txt, float(4),float(45), float(7.5),float(1024), fn_index=1)[0]  #(prompt, samples, steps, scale, seed) #translated_txt
   
   return sd_img_gallery[0]