peterpeter8585 commited on
Commit
de1818a
·
verified ·
1 Parent(s): 4ce3338

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -78,7 +78,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
78
  #m=M.from_pretrained("peterpeter8585/syai4.3")
79
  #t=T.from_pretrained("peterpeter8585/syai4.3")
80
  #pipe=pipeline(model=m,tokenizer=t,task="text-generation")
81
- llm=HuggingFacePipeline.from_model_id(model_id="peterpeter8585/llama3_1",task="text-generation")
 
82
  from langchain.retrievers import WikipediaRetriever as Wiki
83
  import gradio as gr
84
  chatbot = gr.Chatbot(
 
78
  #m=M.from_pretrained("peterpeter8585/syai4.3")
79
  #t=T.from_pretrained("peterpeter8585/syai4.3")
80
  #pipe=pipeline(model=m,tokenizer=t,task="text-generation")
81
+ from langchain_community.llms import GPT4ALL
82
+ llm=GPT4ALL(model="./")
83
  from langchain.retrievers import WikipediaRetriever as Wiki
84
  import gradio as gr
85
  chatbot = gr.Chatbot(