Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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)
|