seawolf2357 commited on
Commit
1f97769
Β·
verified Β·
1 Parent(s): 75c1fd6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -3,6 +3,9 @@ from datasets import load_dataset, Dataset
3
  import faiss # ν•„μš”ν•œ 경우 faissλ₯Ό μž„ν¬νŠΈν•©λ‹ˆλ‹€.
4
  from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
5
  import torch
 
 
 
6
 
7
  # λͺ¨λΈ 및 ν† ν¬λ‚˜μ΄μ € μ„€μ •
8
  model_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
 
3
  import faiss # ν•„μš”ν•œ 경우 faissλ₯Ό μž„ν¬νŠΈν•©λ‹ˆλ‹€.
4
  from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
5
  import torch
6
+ import os
7
+
8
+ tokenkey=os.getenv('HF_API_KEY')
9
 
10
  # λͺ¨λΈ 및 ν† ν¬λ‚˜μ΄μ € μ„€μ •
11
  model_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"