cipherunhsiv commited on
Commit
a6d8a05
·
verified ·
1 Parent(s): 4c9e6b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
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
- !huggingface-cli login
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,