Update model.py
Browse files
model.py
CHANGED
@@ -4,8 +4,8 @@ from typing import Iterator
|
|
4 |
import torch
|
5 |
from transformers import AutoConfig, AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
|
6 |
import os
|
7 |
-
os.environ.getattribute("HF_API_TOKEN")
|
8 |
|
|
|
9 |
|
10 |
model_id = 'Ashishkr/llama2_medical_consultation'
|
11 |
|
|
|
4 |
import torch
|
5 |
from transformers import AutoConfig, AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
|
6 |
import os
|
|
|
7 |
|
8 |
+
token = os.environ.get("HF_API_TOKEN")
|
9 |
|
10 |
model_id = 'Ashishkr/llama2_medical_consultation'
|
11 |
|