my_customisedAgent

Runtime error

Toumaima commited on May 6

Commit

0b2ea5a

verified ·

1 Parent(s): 0a0cd0a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -25,7 +25,7 @@ class BasicAgent:
         api_token = os.getenv("HF_TOKEN")
         os.environ["HUGGINGFACEHUB_API_TOKEN"] = "hf_abcdefghijklmnopqrstuvwxyz"
-        login('HF_TOKEN')
         self.model = InferenceClient("mistralai/Mistral-7B-Instruct-v0.1")
         self.tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.1")  # Initialize tokenizer

         api_token = os.getenv("HF_TOKEN")
         os.environ["HUGGINGFACEHUB_API_TOKEN"] = "hf_abcdefghijklmnopqrstuvwxyz"
+        login(api_token)
         self.model = InferenceClient("mistralai/Mistral-7B-Instruct-v0.1")
         self.tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.1")  # Initialize tokenizer