Spaces:
Runtime error
Runtime error
KvrParaskevi
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,7 @@ from langchain.llms.base import LLM
|
|
4 |
from langchain import PromptTemplate, LLMChain
|
5 |
|
6 |
def initialize_model_and_tokenizer(model_name="KvrParaskevi/Llama-2-7b-Hotel-Booking-Model"):
|
7 |
-
model =
|
8 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
9 |
return model, tokenizer
|
10 |
|
|
|
4 |
from langchain import PromptTemplate, LLMChain
|
5 |
|
6 |
def initialize_model_and_tokenizer(model_name="KvrParaskevi/Llama-2-7b-Hotel-Booking-Model"):
|
7 |
+
model = AutoModelForCausalLLM.from_pretrained(model_name)
|
8 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
9 |
return model, tokenizer
|
10 |
|