Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -41,8 +41,8 @@ h1 {
|
|
41 |
"""
|
42 |
|
43 |
# Load the tokenizer and model
|
44 |
-
tokenizer = AutoTokenizer.from_pretrained("reedmayhew/
|
45 |
-
model = AutoModelForCausalLM.from_pretrained("reedmayhew/
|
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,
|