JayanJames commited on
Commit
16c8adf
·
verified ·
1 Parent(s): 46de224

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -0
app.py CHANGED
@@ -39,3 +39,13 @@ interface = gr.Interface(
39
 
40
  if __name__ == "__main__":
41
  interface.launch()
 
 
 
 
 
 
 
 
 
 
 
39
 
40
  if __name__ == "__main__":
41
  interface.launch()
42
+ import os
43
+ from transformers import AutoTokenizer, AutoModelForCausalLM
44
+
45
+ # Load authentication token from environment variables
46
+ hf_token = os.getenv("HF_AUTH_TOKEN")
47
+
48
+ # Load the Med42 model with the 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)