Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -49,3 +49,13 @@ hf_token = os.getenv("HF_AUTH_TOKEN")
|
|
49 |
med42_model_name = "m42-health/med42-70b"
|
50 |
med42_tokenizer = AutoTokenizer.from_pretrained(med42_model_name, use_auth_token=hf_token)
|
51 |
med42_model = AutoModelForCausalLM.from_pretrained(med42_model_name, use_auth_token=hf_token)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
49 |
med42_model_name = "m42-health/med42-70b"
|
50 |
med42_tokenizer = AutoTokenizer.from_pretrained(med42_model_name, use_auth_token=hf_token)
|
51 |
med42_model = AutoModelForCausalLM.from_pretrained(med42_model_name, use_auth_token=hf_token)
|
52 |
+
import os
|
53 |
+
from transformers import AutoTokenizer, AutoModelForCausalLM
|
54 |
+
|
55 |
+
# Load the token from the environment
|
56 |
+
hf_token = os.getenv("HF_AUTH_TOKEN")
|
57 |
+
|
58 |
+
# Load Med42 model and tokenizer
|
59 |
+
med42_model_name = "m42-health/med42-70b"
|
60 |
+
med42_tokenizer = AutoTokenizer.from_pretrained(med42_model_name, use_auth_token=hf_token)
|
61 |
+
med42_model = AutoModelForCausalLM.from_pretrained(med42_model_name, use_auth_token=hf_token)
|