Spaces:

debisoft
/

tsuin-complete

Running on Zero

debisoft commited on Feb 24

Commit

d9bc952

1 Parent(s): 43f6867

login

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 @spaces.GPU
 def sentience_check():
-    huggingface_hub.login(token=os.getenv('HUGGINGFACE_TOKEN'))
     device = torch.device("cuda")
     tokenizer = AutoTokenizer.from_pretrained("debisoft/gemma-2-2B-it-thinking-function_calling-V0")
     model = AutoModelForCausalLM.from_pretrained("debisoft/gemma-2-2B-it-thinking-function_calling-V0").to(device)

 @spaces.GPU
 def sentience_check():
+    huggingface_hub.login(os.getenv('HUGGINGFACE_TOKEN'))
     device = torch.device("cuda")
     tokenizer = AutoTokenizer.from_pretrained("debisoft/gemma-2-2B-it-thinking-function_calling-V0")
     model = AutoModelForCausalLM.from_pretrained("debisoft/gemma-2-2B-it-thinking-function_calling-V0").to(device)