Update app.py
Browse files
app.py
CHANGED
@@ -21,7 +21,7 @@ st.write("I am your hotel booking assistant for today.")
|
|
21 |
tokenizer, model = demo_chat.load_model()
|
22 |
|
23 |
model_identifier = "KvrParaskevi/Hotel-Assistant-Attempt4-Llama-2-7b"
|
24 |
-
task = "text-
|
25 |
|
26 |
# Load the model using the pipeline
|
27 |
model_pipeline = pipeline(task, model=model,tokenizer=tokenizer)
|
|
|
21 |
tokenizer, model = demo_chat.load_model()
|
22 |
|
23 |
model_identifier = "KvrParaskevi/Hotel-Assistant-Attempt4-Llama-2-7b"
|
24 |
+
task = "text-generation" # Change this to your model's task
|
25 |
|
26 |
# Load the model using the pipeline
|
27 |
model_pipeline = pipeline(task, model=model,tokenizer=tokenizer)
|