Manasa1 commited on
Commit
f321fb1
·
verified ·
1 Parent(s): 4e6b33c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -7,11 +7,8 @@ import gradio as gr
7
  from transformers import AutoTokenizer, AutoModelForCausalLM
8
  from huggingface_hub import hf_hub_download
9
 
10
- def load_model():
11
- model_name = 'TheBloke/Llama-2-7B-Chat-GGML'
12
- tokenizer = AutoTokenizer.from_pretrained(model_name)
13
- model = AutoModelForCausalLM.from_pretrained(model_name)
14
- return tokenizer, model
15
  DB_FAISS_PATH = "vectorstores/db_faiss"
16
 
17
  custom_prompt_template = """Use the following pieces of information to answer the user's question.
 
7
  from transformers import AutoTokenizer, AutoModelForCausalLM
8
  from huggingface_hub import hf_hub_download
9
 
10
+
11
+
 
 
 
12
  DB_FAISS_PATH = "vectorstores/db_faiss"
13
 
14
  custom_prompt_template = """Use the following pieces of information to answer the user's question.