Add USER_TOKEN

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import torch
 from diffusers import DiffusionPipeline
+import streamlit as st
 from transformers import (
     pipeline,
     MBart50TokenizerFast,
@@ -23,8 +24,9 @@ trans_model = MBartForConditionalGeneration.from_pretrained("facebook/mbart-larg
 model_id = "CompVis/stable-diffusion-v1-4"
 
 pipe = DiffusionPipeline.from_pretrained(
-    "CompVis/stable-diffusion-v1-4",
+    model_id,
     custom_pipeline="multilingual_stable_diffusion",
+    use_auth_token=st.secrets["USER_TOKEN"],
     detection_pipeline=language_detection_pipeline,
     translation_model=trans_model,
     translation_tokenizer=trans_tokenizer,