Mohamed284 commited on
Commit
5546056
·
verified ·
1 Parent(s): 7e9c748

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -30,8 +30,12 @@ CHUNK_SIZE = 800
30
  OVERLAP = 200
31
  EMBEDDING_BATCH_SIZE = 32 # Batch size for embedding API calls
32
 
33
- # Login to Hugging Face Hub
34
- login(token="llama3")
 
 
 
 
35
 
36
  # Initialize models
37
  embedding_model = SentenceTransformer(embedding_model_name)
 
30
  OVERLAP = 200
31
  EMBEDDING_BATCH_SIZE = 32 # Batch size for embedding API calls
32
 
33
+ hf_token = os.environ.get('HF_TOKEN')
34
+ if hf_token:
35
+ login(token=hf_token)
36
+ else:
37
+ raise ValueError("Hugging Face token not found in environment variables")
38
+
39
 
40
  # Initialize models
41
  embedding_model = SentenceTransformer(embedding_model_name)