Tanmay211998 commited on
Commit
3ef45f6
·
verified ·
1 Parent(s): 87cc506

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -4,6 +4,7 @@
4
  from llama_index.core import VectorStoreIndex,SimpleDirectoryReader,ServiceContext
5
  from llama_index.llms.huggingface import HuggingFaceLLM
6
  from llama_index.core.prompts.prompts import SimpleInputPrompt
 
7
 
8
  documents = SimpleDirectoryReader('./pdf').load_data()
9
  documents
@@ -16,8 +17,9 @@ accurately as possible based on the instructions and context provided.
16
  query_wrapper_prompt = SimpleInputPrompt("<|USER|>{query_str}<|ASSISTANT|>")
17
 
18
  #!!huggingface-cli login
19
- huggingface-cli login --token HFToken
20
-
 
21
 
22
 
23
  import torch
 
4
  from llama_index.core import VectorStoreIndex,SimpleDirectoryReader,ServiceContext
5
  from llama_index.llms.huggingface import HuggingFaceLLM
6
  from llama_index.core.prompts.prompts import SimpleInputPrompt
7
+ import subprocess
8
 
9
  documents = SimpleDirectoryReader('./pdf').load_data()
10
  documents
 
17
  query_wrapper_prompt = SimpleInputPrompt("<|USER|>{query_str}<|ASSISTANT|>")
18
 
19
  #!!huggingface-cli login
20
+ # Run the command to login using the token
21
+ command = f"huggingface-cli login --token {HFToken}"
22
+ subprocess.run(command, shell=True)
23
 
24
 
25
  import torch