Spaces:
Build error
Build error
Update app.py
Browse files
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 |
-
|
| 34 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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)
|