Michael Bernovskiy commited on
Commit
139887d
1 Parent(s): 3cf2b77

concrete model

Browse files
Files changed (1) hide show
  1. backend/query_llm.py +2 -6
backend/query_llm.py CHANGED
@@ -7,16 +7,12 @@ from typing import Any, Dict, Generator, List
7
  from huggingface_hub import InferenceClient
8
  from transformers import AutoTokenizer
9
 
10
- from dotenv import load_dotenv
11
-
12
-
13
- load_dotenv()
14
-
15
 
16
 
17
  OPENAI_KEY = os.getenv("OPENAI_API_KEY")
18
  HF_TOKEN = os.getenv("HF_TOKEN")
19
- TOKENIZER = AutoTokenizer.from_pretrained(os.getenv("HF_MODEL"))
 
20
 
21
  HF_CLIENT = InferenceClient(
22
  os.getenv("HF_MODEL"),
 
7
  from huggingface_hub import InferenceClient
8
  from transformers import AutoTokenizer
9
 
 
 
 
 
 
10
 
11
 
12
  OPENAI_KEY = os.getenv("OPENAI_API_KEY")
13
  HF_TOKEN = os.getenv("HF_TOKEN")
14
+ model_name = 'mistralai/Mistral-7B-Instruct-v0.2'
15
+ TOKENIZER = AutoTokenizer.from_pretrained(model_name)
16
 
17
  HF_CLIENT = InferenceClient(
18
  os.getenv("HF_MODEL"),