Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -6,6 +6,7 @@ from llama_index.llms.huggingface import HuggingFaceLLM
|
|
| 6 |
from llama_index.core.prompts.prompts import SimpleInputPrompt
|
| 7 |
import subprocess
|
| 8 |
import os
|
|
|
|
| 9 |
|
| 10 |
documents = SimpleDirectoryReader('./pdf').load_data()
|
| 11 |
documents
|
|
@@ -21,6 +22,9 @@ query_wrapper_prompt = SimpleInputPrompt("<|USER|>{query_str}<|ASSISTANT|>")
|
|
| 21 |
# Run the command to login using the token
|
| 22 |
VarHFToken = os.getenv("HFToken")
|
| 23 |
print(VarHFToken)
|
|
|
|
|
|
|
|
|
|
| 24 |
#command = f"huggingface-cli login --token {VarHFToken}"
|
| 25 |
#subprocess.run(command, shell=True)
|
| 26 |
|
|
|
|
| 6 |
from llama_index.core.prompts.prompts import SimpleInputPrompt
|
| 7 |
import subprocess
|
| 8 |
import os
|
| 9 |
+
from huggingface_hub import login
|
| 10 |
|
| 11 |
documents = SimpleDirectoryReader('./pdf').load_data()
|
| 12 |
documents
|
|
|
|
| 22 |
# Run the command to login using the token
|
| 23 |
VarHFToken = os.getenv("HFToken")
|
| 24 |
print(VarHFToken)
|
| 25 |
+
login(VarHFToken)
|
| 26 |
+
|
| 27 |
+
|
| 28 |
#command = f"huggingface-cli login --token {VarHFToken}"
|
| 29 |
#subprocess.run(command, shell=True)
|
| 30 |
|