test

app.py

@@ -7,6 +7,10 @@ from torch import autocast
 from diffusers import StableDiffusionPipeline, DDIMScheduler,  DiffusionPipeline
 import streamlit as st
 
+from huggingface_hub import login
+login()
+
+
 # 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")