Update app.py

app.py

@@ -37,16 +37,19 @@ for message in messages:
 
 # Get user input
 user_input = st.text_input("Your message")
+print("received!")
 
 # Generate response
 if user_input:
     messages.append({"role": "user", "content": user_input})
+    print("good!")
     text = tokenizer.apply_chat_template(
         messages,
         tokenize=False,
         add_generation_prompt=True
     )
     model_inputs = tokenizer([text], return_tensors="pt").to(device)
+    print("good!")
     generated_ids = model.generate(
         model_inputs.input_ids,
         max_new_tokens=512
@@ -54,7 +57,9 @@ if user_input:
     generated_ids = [
         output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)
     ]
+    print("good!")
     response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
+    print("good!")
     messages.append({"role": "assistant", "content": response})
 
     # Display response