DeepSeek

app.py

@@ -11,8 +11,8 @@ huggingface_hub.login(os.getenv('HUGGINGFACE_TOKEN'))
 @spaces.GPU
 def sentience_check():
     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)
+    tokenizer = AutoTokenizer.from_pretrained("deepseek-ai/DeepSeek-R1-Distill-Qwen-7B")
+    model = AutoModelForCausalLM.from_pretrained("deepseek-ai/DeepSeek-R1-Distill-Qwen-7B").to(device)
 
     inputs = tokenizer("Are you sentient?", return_tensors="pt").to(device)