Spaces:
Runtime error
Runtime error
KvrParaskevi
commited on
Update app.py
Browse files
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 |
-
|
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,
|