zerostratos commited on
Commit
e976d8b
·
verified ·
1 Parent(s): f104694

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ llm = Llama.from_pretrained(
8
  )
9
  # Define the function to interact with the model
10
  def chat_with_model(user_input):
11
- response = llm.generate(
12
  messages=[
13
  {"role": "user", "content": user_input}
14
  ]
 
8
  )
9
  # Define the function to interact with the model
10
  def chat_with_model(user_input):
11
+ response = llm.create_chat_completion(
12
  messages=[
13
  {"role": "user", "content": user_input}
14
  ]