Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -26,9 +26,10 @@ def mod(pdf_url):
|
|
| 26 |
accurately as possible based on the instructions and context provided."""
|
| 27 |
query_wrapper_prompt = SimpleInputPrompt("<|USER|>{query_str}<|ASSISTANT|>")
|
| 28 |
from huggingface_hub import login
|
| 29 |
-
|
| 30 |
-
|
| 31 |
-
|
|
|
|
| 32 |
llm = HuggingFaceLLM(
|
| 33 |
context_window=4096,
|
| 34 |
max_new_tokens=750,
|
|
|
|
| 26 |
accurately as possible based on the instructions and context provided."""
|
| 27 |
query_wrapper_prompt = SimpleInputPrompt("<|USER|>{query_str}<|ASSISTANT|>")
|
| 28 |
from huggingface_hub import login
|
| 29 |
+
hf_token = os.environ.get('HF_TOKEN')
|
| 30 |
+
if not hf_token:
|
| 31 |
+
raise ValueError("HF_TOKEN environment variable not found. Please set it in your Space settings.")
|
| 32 |
+
login(token=hf_token)
|
| 33 |
llm = HuggingFaceLLM(
|
| 34 |
context_window=4096,
|
| 35 |
max_new_tokens=750,
|