Update app.py

app.py

@@ -5,15 +5,12 @@ import base64
 import gradio as gr
 import os
 from gradio.components import Textbox
-from huggingface_hub import notebook_login
-import ipywidgets
-
-notebook_login()
 
 HF_TOKEN = os.getenv('HF_TOKEN')
 hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "movie-img-generator")
+auth_token = os.environ.get("auth_token")
 
-pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=HF_TOKEN)
+pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=auth_token)
 pipe = pipe.to("cuda")
 
 device = "cuda"