Changes

app.py

@@ -7,11 +7,11 @@ from torch import autocast
 from diffusers import StableDiffusionPipeline, DDIMScheduler,  DiffusionPipeline
 import streamlit as st
 
-# from huggingface_hub import login
+from huggingface_hub import login
 
 # HF_TOKEN = os.environ.get("HF_TOKEN")
 
-# login(token  = HF_TOKEN)
+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