umaiku commited on
Commit
60ac7f7
·
verified ·
1 Parent(s): f2e8f60

Update app.py

Browse files

Change to Mistral

Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -9,8 +9,11 @@ import os
9
  For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
10
  """
11
  login(token=os.getenv('TOKEN'))
12
- client = InferenceClient("meta-llama/Llama-3.2-1B-Instruct")
13
- #client = InferenceClient("mistralai/Mistral-7B-Instruct-v0.3")
 
 
 
14
 
15
  folder = snapshot_download(repo_id="umaiku/faiss_index", repo_type="dataset", local_dir=os.getcwd())
16
 
 
9
  For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
10
  """
11
  login(token=os.getenv('TOKEN'))
12
+
13
+ #model = "meta-llama/Llama-3.2-1B-Instruct"
14
+ model = "mistralai/Mistral-7B-Instruct-v0.3"
15
+
16
+ client = InferenceClient(model)
17
 
18
  folder = snapshot_download(repo_id="umaiku/faiss_index", repo_type="dataset", local_dir=os.getcwd())
19