KvrParaskevi commited on
Commit
548127e
·
verified ·
1 Parent(s): 0a0731c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -12,11 +12,9 @@ def initialize_model_and_tokenizer(model_name="KvrParaskevi/Llama-2-7b-Hotel-Boo
12
  tokenizer = AutoTokenizer.from_pretrained(model_name)
13
  return model, tokenizer
14
 
15
- model, tokenizer = initialize_model_and_tokenizer()
16
-
17
  @spaces.GPU
18
  def load_pipeline():
19
- tokenizer, model = load_model()
20
  pipe = pipeline("text-generation",
21
  model= model,
22
  tokenizer = tokenizer,
 
12
  tokenizer = AutoTokenizer.from_pretrained(model_name)
13
  return model, tokenizer
14
 
 
 
15
  @spaces.GPU
16
  def load_pipeline():
17
+ model, tokenizer = initialize_model_and_tokenizer()
18
  pipe = pipeline("text-generation",
19
  model= model,
20
  tokenizer = tokenizer,