Toumaima commited on
Commit
c66f031
·
verified ·
1 Parent(s): c53f5e5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -26,9 +26,7 @@ class BasicAgent:
26
 
27
  os.environ["HUGGINGFACEHUB_API_TOKEN"] = "hf_abcdefghijklmnopqrstuvwxyz"
28
  login(api_token)
29
- password = os.environ.get(api_token)
30
- if password is None:
31
- password = input('Please enter your password: ')
32
  self.model = InferenceClient("mistralai/Mistral-7B-Instruct-v0.1")
33
  self.tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.1") # Initialize tokenizer
34
  SYSTEM_PROMPT = """You are a general AI assistant. I will ask you a question. Report your thoughts, and
 
26
 
27
  os.environ["HUGGINGFACEHUB_API_TOKEN"] = "hf_abcdefghijklmnopqrstuvwxyz"
28
  login(api_token)
29
+ print(f"Using API Token: {api_token}")
 
 
30
  self.model = InferenceClient("mistralai/Mistral-7B-Instruct-v0.1")
31
  self.tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.1") # Initialize tokenizer
32
  SYSTEM_PROMPT = """You are a general AI assistant. I will ask you a question. Report your thoughts, and