Update app.py

app.py

@@ -19,7 +19,7 @@ model = AutoModelForCausalLM.from_pretrained(
 
 
 st.title("💬 Chatbot")
-st.caption("🚀 A streamlit chatbot powered by Microsoft Phi-3-mini")
+st.caption("🚀 A streamlit chatbot powered by Google's Gemma")
 
 # Initialize chat history
 if 'messages' not in st.session_state:
@@ -46,7 +46,9 @@ if prompt := st.chat_input():
     inputs = tokenizer.encode(text, add_special_tokens=False, return_tensors="pt")
     outputs = model.generate(input_ids=inputs.to(model.device), max_new_tokens=150)
 
-    msg = outputs #output[0]['generated_text']
+    
+
+    msg = tokenizer.decode(outputs[0]) #output[0]['generated_text']
     
      # Display assistant response in chat message container
     st.chat_message("assistant").write(msg)