Spaces:

LLM-DEMO
/

llama-2-medical-consultation

Paused

Ashishkr commited on Aug 24, 2023

Commit

0da0787

•

1 Parent(s): 5d42eda

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -4,8 +4,8 @@ from typing import Iterator
 import torch
 from transformers import AutoConfig, AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
 import os
-os.environ.getattribute("HF_API_TOKEN")
 model_id = 'Ashishkr/llama2_medical_consultation'

 import torch
 from transformers import AutoConfig, AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
 import os
+token = os.environ.get("HF_API_TOKEN")
 model_id = 'Ashishkr/llama2_medical_consultation'