mery22 commited on
Commit
77bd906
1 Parent(s): 922a129

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -6,6 +6,7 @@ from transformers import (
6
  BitsAndBytesConfig,
7
  pipeline
8
  )
 
9
 
10
  from transformers import BitsAndBytesConfig
11
  from langchain.embeddings.huggingface import HuggingFaceEmbeddings
@@ -23,7 +24,7 @@ from transformers import pipeline
23
  import transformers
24
  model_name='mistralai/Mistral-7B-Instruct-v0.1'
25
  from huggingface_hub import login
26
- login(token=st.secrets["HF_TOKEN"])
27
  model_config = transformers.AutoConfig.from_pretrained(
28
  model_name,
29
  )
 
6
  BitsAndBytesConfig,
7
  pipeline
8
  )
9
+ import gradio as gr
10
 
11
  from transformers import BitsAndBytesConfig
12
  from langchain.embeddings.huggingface import HuggingFaceEmbeddings
 
24
  import transformers
25
  model_name='mistralai/Mistral-7B-Instruct-v0.1'
26
  from huggingface_hub import login
27
+ login(token=gr.secrets["HF_TOKEN"])
28
  model_config = transformers.AutoConfig.from_pretrained(
29
  model_name,
30
  )