Update app.py
Browse files
app.py
CHANGED
@@ -66,8 +66,8 @@ from langchain.chains import create_retrieval_chain
|
|
66 |
#from langchain_community.llms import Ollama
|
67 |
#READER_MODEL=Ollama(model="meta-llama/Meta-Llama-Guard-2-8B")
|
68 |
|
69 |
-
from transformers import
|
70 |
-
lll_model =
|
71 |
|
72 |
#READER_MODEL = "HuggingFaceH4/zephyr-7b-beta"
|
73 |
|
|
|
66 |
#from langchain_community.llms import Ollama
|
67 |
#READER_MODEL=Ollama(model="meta-llama/Meta-Llama-Guard-2-8B")
|
68 |
|
69 |
+
from transformers import AutoModelForCausalLM
|
70 |
+
lll_model = AutoModelForCausalLM.from_pretrained("HuggingFaceH4/zephyr-7b-beta")
|
71 |
|
72 |
#READER_MODEL = "HuggingFaceH4/zephyr-7b-beta"
|
73 |
|