reedmayhew commited on
Commit
b78d721
·
verified ·
1 Parent(s): 8cd3c65

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -41,8 +41,8 @@ h1 {
41
  """
42
 
43
  # Load the tokenizer and model
44
- tokenizer = AutoTokenizer.from_pretrained("reedmayhew/DeepSeek-R1-Refined-Llama-3.1-8B-hf", device_map="cuda")
45
- model = AutoModelForCausalLM.from_pretrained("reedmayhew/DeepSeek-R1-Refined-Llama-3.1-8B-hf", device_map="cuda")
46
 
47
  terminators = [
48
  tokenizer.eos_token_id,
 
41
  """
42
 
43
  # Load the tokenizer and model
44
+ tokenizer = AutoTokenizer.from_pretrained("reedmayhew/HealthCare-Reasoning-Assistant-Llama-3.1-8B-HF", device_map="cuda")
45
+ model = AutoModelForCausalLM.from_pretrained("reedmayhew/HealthCare-Reasoning-Assistant-Llama-3.1-8B-HF", device_map="cuda")
46
 
47
  terminators = [
48
  tokenizer.eos_token_id,