Hugging Face's logo

Spaces:
DMID23
/
Spindle-LLM
Sleeping

App Files Community
DMID23 commited on
Commit
f92e66d
·
verified ·
1 Parent(s): 493d449

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -4,10 +4,9 @@ import os
4
  import subprocess
5
  import torch
6
  import huggingface_hub
7
- from transformers import AutoModelForCausalLM, AutoTokenizer
8
-
9
- huggingface_hub.login("your_token_here")
10
 
 
 
11
 
12
  base_model = "mistralai/Mistral-7B-Instruct-v0.1"
13
 
 
4
  import subprocess
5
  import torch
6
  import huggingface_hub
 
 
 
7
 
8
+ token = os.environ.get("HUGGING_FACE_HUB_TOKEN")
9
+ login(token)
10
 
11
  base_model = "mistralai/Mistral-7B-Instruct-v0.1"
12