Biswa000 commited on
Commit
5139ba3
·
verified ·
1 Parent(s): 293db5c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -11,6 +11,13 @@ import torch
11
  from sentence_transformers import SentenceTransformer
12
  from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig
13
 
 
 
 
 
 
 
 
14
  # ---------- CONFIG ----------
15
  MODEL_ID = "mistralai/Mistral-7B-Instruct-v0.3" # change if needed
16
  EMBED_MODEL = "sentence-transformers/all-MiniLM-L6-v2"
 
11
  from sentence_transformers import SentenceTransformer
12
  from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig
13
 
14
+
15
+ from huggingface_hub import login
16
+ import os
17
+
18
+ login(os.getenv("HF_TOKEN"))
19
+
20
+
21
  # ---------- CONFIG ----------
22
  MODEL_ID = "mistralai/Mistral-7B-Instruct-v0.3" # change if needed
23
  EMBED_MODEL = "sentence-transformers/all-MiniLM-L6-v2"