Zwea Htet commited on
Commit
66bfc6b
1 Parent(s): 9fb0f7d

added huggingface api

Browse files
Files changed (1) hide show
  1. models/vector_database.py +12 -2
models/vector_database.py CHANGED
@@ -8,7 +8,7 @@ from llama_index.core import (
8
  VectorStoreIndex,
9
  StorageContext,
10
  )
11
- from huggingface_hub import HfFileSystem
12
 
13
  import os
14
 
@@ -70,4 +70,14 @@ def get_pinecone_index(filename: str) -> VectorStoreIndex:
70
  documents=docs, show_progress=True, storage_context=storage_context
71
  )
72
 
73
- return index
 
 
 
 
 
 
 
 
 
 
 
8
  VectorStoreIndex,
9
  StorageContext,
10
  )
11
+ from huggingface_hub import HfFileSystem, HfApi
12
 
13
  import os
14
 
 
70
  documents=docs, show_progress=True, storage_context=storage_context
71
  )
72
 
73
+ return index
74
+
75
+ api = HfApi(
76
+ token=os.environ.get("HF_TOKEN")
77
+ )
78
+
79
+ api.upload_file(
80
+ repo_id="hbui/RegBot4.0",
81
+ path_or_fileobj=f"{SAVE_DIR}/calregs.pdf",
82
+ path_in_repo=f"{SAVE_DIR}/calregs.pdf",
83
+ )