JayanJames commited on
Commit
73129a4
·
verified ·
1 Parent(s): 16c8adf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -0
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)