Update app.py

app.py

@@ -1,12 +1,15 @@
 import torch
 import streamlit as st
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+# Define SystemMessage class if needed
+class SystemMessage:
+    def __init__(self, content):
+        self.content = content
 
-system_message = "Welcome to my Chatbot! Ask me anything."
-
-# Display the message using Streamlit's `write` function
+system_message = SystemMessage("Welcome to my Chatbot! Ask me anything.")
 st.write(system_message)
 
+
 # Choose your desired free model from the Hugging Face Hub
 model_name = "t5-small"  # Replace with your choice (e.g., facebook/bart-base or EleutherAI/gpt-neo-125M)