Changes

app.py

@@ -8,14 +8,17 @@ from diffusers import StableDiffusionPipeline, DDIMScheduler,  DiffusionPipeline
 import streamlit as st
 
 from huggingface_hub import login
-login(token  = "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")
+
+HF_TOKEN = os.environ.get("HF_TOKEN")
+
+login(token  = HF_TOKEN)
 
 
 # model_path = WEIGHTS_DIR             # If you want to use previously trained model saved in gdrive, replace this with the full path of model in gdrive
 # headers = {"Authorization": "Bearer xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"}
-# pipe = StableDiffusionPipeline.from_pretrained("Divyanshu04/Finetuned-sd-vae", safety_checker=None, torch_dtype=torch.float32).to("cuda")
 
-pipe = StableDiffusionPipeline.from_pretrained("Divyanshu04/Finetuned-model", torch_dtype=torch.float16,  use_safetensors=True, use_auth_token = "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")
+pipe = StableDiffusionPipeline.from_pretrained("Divyanshu04/Finetuned-model", safety_checker=None, torch_dtype=torch.float32).to("cuda")
+
 pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
 pipe.enable_xformers_memory_efficient_attention()
 g_cuda = None