Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -30,7 +30,7 @@ from langchain_huggingface import HuggingFacePipeline
|
|
| 30 |
|
| 31 |
# Optional: Set HF Token if needed-allWrite
|
| 32 |
os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.getenv("HF_TOKEN")
|
| 33 |
-
|
| 34 |
# Initialize embedding model
|
| 35 |
embedding_model = HuggingFaceEmbeddings(model_name="all-MiniLM-L6-v2")
|
| 36 |
|
|
@@ -249,6 +249,7 @@ def upload_and_prepare(file,user):
|
|
| 249 |
else:
|
| 250 |
mm="❌ Unauthorized User"
|
| 251 |
# After successful upload
|
|
|
|
| 252 |
pdf_files = api.list_repo_files(repo_id, repo_type="dataset")
|
| 253 |
pdf_links = "\n".join([f"• [📄 {f}](https://huggingface.co/datasets/{repo_id}/resolve/main/{f})"
|
| 254 |
for f in pdf_files if f.endswith('.pdf')])
|
|
|
|
| 30 |
|
| 31 |
# Optional: Set HF Token if needed-allWrite
|
| 32 |
os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.getenv("HF_TOKEN")
|
| 33 |
+
api = HfApi(token=os.getenv("HF_TOKEN")) # Replace with your token
|
| 34 |
# Initialize embedding model
|
| 35 |
embedding_model = HuggingFaceEmbeddings(model_name="all-MiniLM-L6-v2")
|
| 36 |
|
|
|
|
| 249 |
else:
|
| 250 |
mm="❌ Unauthorized User"
|
| 251 |
# After successful upload
|
| 252 |
+
api = HfApi(token=os.getenv("HF_TOKEN")) # Replace with your token
|
| 253 |
pdf_files = api.list_repo_files(repo_id, repo_type="dataset")
|
| 254 |
pdf_links = "\n".join([f"• [📄 {f}](https://huggingface.co/datasets/{repo_id}/resolve/main/{f})"
|
| 255 |
for f in pdf_files if f.endswith('.pdf')])
|