manabb commited on
Commit
8208e45
·
verified ·
1 Parent(s): 66455d8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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')])