update

app.py

@@ -12,16 +12,14 @@ 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",
-    use_auth_token= HF_TOKEN).to(device)
+pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", revision="fp16", torch_dtype=torch.float16, use_auth_token= HF_TOKEN).to(device)
 
 def get_sd(translated_txt):
     scale=7.5 
     steps=45
     with autocast('cpu' if not th.cuda.is_available() else 'cuda'):
         image = pipe(translated_txt, guidance_scale=scale, num_inference_steps=steps)["sample"][0]
-        return image
+    return image
 
 #API_URL = "https://api-inference.huggingface.co/models/bigscience/bloom"
 #HF_TOKEN = os.environ.get("diffuse_new") or True