Update app.py

app.py

@@ -26,8 +26,8 @@ tokenizer = None
 def initialize_ai_model():
     global model, tokenizer
     try:
-        print("🤖 Loading BitNet AI model for personalized welcomes...")
-        model_id = "microsoft/bitnet-b1.58-2B-4T"
+        print("🤖 Loading SmolLM2 AI model for personalized welcomes...")
+        model_id = "HuggingFaceTB/SmolLM2-360M-Instruct"
         tokenizer = AutoTokenizer.from_pretrained(model_id)
         model = AutoModelForCausalLM.from_pretrained(
             model_id,
@@ -62,7 +62,7 @@ def store_full_introduction(message_content, author_name):
     }
 
 async def generate_ai_welcome_message(new_members, recent_context):
-    """Generate a personalized welcome message using BitNet AI with full introduction context"""
+    """Generate a personalized welcome message using SmolLM2 AI with full introduction context"""
     if not model or not tokenizer:
         return None
     
@@ -268,6 +268,6 @@ demo = gr.Interface(
     inputs="text", 
     outputs="text",
     title="🤖 AI-Powered Discord Welcome Bot",
-    description="Enhanced with BitNet b1.58 for personalized community welcomes!"
+    description="Enhanced with SmolLM2-360M-Instruct for personalized community welcomes!"
 )
-demo.launch()
+demo.launch()