KvrParaskevi commited on
Commit
857f215
·
verified ·
1 Parent(s): 046f5e1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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-classification" # Change this to your model's task
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)