Fiqa commited on
Commit
90c9be8
·
verified ·
1 Parent(s): 8b3ca46

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -3,6 +3,7 @@ from huggingface_hub import login
3
  from transformers import BlipProcessor, BlipForConditionalGeneration
4
  from transformers import MllamaForConditionalGeneration, AutoProcessor
5
  from PIL import Image
 
6
 
7
  import gradio as gr
8
  from diffusers import DiffusionPipeline
@@ -25,7 +26,10 @@ textile_designs = ['woven texture', 'embroidery', 'printed fabric', 'hand-dyed',
25
 
26
 
27
  # Get Hugging Face Token from environment variable
28
- hf_token = os.getenv('HF_AUTH_TOKEN')
 
 
 
29
  if not hf_token:
30
  raise ValueError("Hugging Face token is not set in the environment variables.")
31
  login(token=hf_token)
 
3
  from transformers import BlipProcessor, BlipForConditionalGeneration
4
  from transformers import MllamaForConditionalGeneration, AutoProcessor
5
  from PIL import Image
6
+ from dotenv import load_dotenv
7
 
8
  import gradio as gr
9
  from diffusers import DiffusionPipeline
 
26
 
27
 
28
  # Get Hugging Face Token from environment variable
29
+ HUGGING_FACE_TOKEN = os.getenv("HUGGING_FACE_TOKEN")
30
+
31
+ # Authenticate using the token
32
+ login(HUGGING_FACE_TOKEN)
33
  if not hf_token:
34
  raise ValueError("Hugging Face token is not set in the environment variables.")
35
  login(token=hf_token)