KvrParaskevi commited on
Commit
950ec0e
·
verified ·
1 Parent(s): 85a3c90

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = AutoModelForCausalLM.from_pretrained(model_name)
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