MD1998 commited on
Commit
d0a604a
·
verified ·
1 Parent(s): e35a322

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -1,15 +1,13 @@
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",
 
1
  import gradio as gr
2
+ from langchain.llms import HuggingFacePipeline
3
  from transformers import AutoTokenizer, AutoModel
4
  import transformers
5
  import torch
6
  import warnings
7
  warnings.filterwarnings('ignore')
8
 
9
+ model = 'google/flan-t5-base'
10
+ tokenizer=AutoTokenizer.from_pretrained(model)
 
 
11
 
12
  pipeline=transformers.pipeline(
13
  "text-generation",