MD1998 commited on
Commit
6c68fe0
·
verified ·
1 Parent(s): 2e53cf6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -1,18 +1,19 @@
1
  import gradio as gr
2
- from langchain.llms import HuggingFacePipeline
3
- from transformers import AutoTokenizer
4
  import transformers
5
  import torch
6
  import warnings
7
  warnings.filterwarnings('ignore')
8
 
9
- # llm = CTransformers(model="TheBloke/Llama-2-7B-Chat-GGML", model_file="llama-2-7b-chat.ggmlv3.q3_K_S.bin")
 
10
 
11
  tokenizer=AutoTokenizer.from_pretrained(model="TheBloke/Llama-2-7B-Chat-GGML", model_file="llama-2-7b-chat.ggmlv3.q3_K_S.bin")
12
 
13
  pipeline=transformers.pipeline(
14
  "text-generation",
15
- model=(model="TheBloke/Llama-2-7B-Chat-GGML", model_file="llama-2-7b-chat.ggmlv3.q3_K_S.bin"),
16
  tokenizer=tokenizer,
17
  torch_dtype=torch.bfloat16,
18
  trust_remote_code=True,
 
1
  import gradio as gr
2
+ from langchain.llms import HuggingFacePipeline, CTransformers
3
+ from transformers import AutoTokenizer, AutoModel
4
  import transformers
5
  import torch
6
  import warnings
7
  warnings.filterwarnings('ignore')
8
 
9
+ model = CTransformers(model="TheBloke/Llama-2-7B-Chat-GGML", model_file="llama-2-7b-chat.ggmlv3.q3_K_S.bin")
10
+
11
 
12
  tokenizer=AutoTokenizer.from_pretrained(model="TheBloke/Llama-2-7B-Chat-GGML", model_file="llama-2-7b-chat.ggmlv3.q3_K_S.bin")
13
 
14
  pipeline=transformers.pipeline(
15
  "text-generation",
16
+ model=model,
17
  tokenizer=tokenizer,
18
  torch_dtype=torch.bfloat16,
19
  trust_remote_code=True,