debisoft commited on
Commit
2ad699d
·
1 Parent(s): d9bc952

login outside

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -6,9 +6,10 @@ import torch
6
 
7
  from transformers import AutoTokenizer, AutoModelForCausalLM
8
 
 
 
9
  @spaces.GPU
10
  def sentience_check():
11
- huggingface_hub.login(os.getenv('HUGGINGFACE_TOKEN'))
12
  device = torch.device("cuda")
13
  tokenizer = AutoTokenizer.from_pretrained("debisoft/gemma-2-2B-it-thinking-function_calling-V0")
14
  model = AutoModelForCausalLM.from_pretrained("debisoft/gemma-2-2B-it-thinking-function_calling-V0").to(device)
 
6
 
7
  from transformers import AutoTokenizer, AutoModelForCausalLM
8
 
9
+ huggingface_hub.login(os.getenv('HUGGINGFACE_TOKEN'))
10
+
11
  @spaces.GPU
12
  def sentience_check():
 
13
  device = torch.device("cuda")
14
  tokenizer = AutoTokenizer.from_pretrained("debisoft/gemma-2-2B-it-thinking-function_calling-V0")
15
  model = AutoModelForCausalLM.from_pretrained("debisoft/gemma-2-2B-it-thinking-function_calling-V0").to(device)