s

ChatAI/chat_ai.py

@@ -1,4 +1,4 @@
 from transformers import pipeline
 
 
-pipe = pipeline("text2text-generation", model="google/flan-t5-xxl")
+pipe = pipeline("text-generation", model="microsoft/phi-2")

app.py

@@ -2,33 +2,33 @@ import discord
 from discord.ext import commands
 from ChatAI.chat_ai import pipe as ai
 
+
 # Set up Discord bot intents and command prefix
 intents = discord.Intents.default()
 intents.message_content = True
 intents.messages = True
 bot = commands.Bot(command_prefix="!", intents=intents)
-
 # Dictionary to track message count per channel
 message_counts = {}
 
 @bot.event
 async def on_message(message):
     guild = message.guild  # Get the guild (server) the message is from
-    channel = message.channel
-    if message.channel != discord.utils.get(guild.text_channels, name="ai_chatter"):
-        print("not ai_chatter channel"); return;
-    print("channel: ai_chatter")
+    channel = discord.utils.get(guild.text_channels, name="ai_chatter")
 
-    if message.author.bot:
-        print("not taking own messages into count"); return;
+    # Verification
+    if message.channel != channel: return;
+    if message.author.bot: return;
 
-    if message.author.name!="sandra_n": pass;
-    print("I can see its you, sis"); await message.channel.send("I can see its you, sis")
+    # Per user
+    x = guild.members
+    if message.author.name=="sandra_n": await message.channel.send("I can see its you, sis");
 
-    if message.channel.id not in message_counts: # Ensure tracking exists for this channel
+    # Ensure tracking exists for this channel
+    if message.channel.id not in message_counts:
         message_counts[message.channel.id] = 0
-
-    message_counts[message.channel.id] += 1 # Increment message count
+    # Increment message count
+    message_counts[message.channel.id] += 1
     print(message_counts[message.channel.id])
 
     messages = []

requirements.txt

@@ -2,4 +2,4 @@ transformers
 huggingface_hub==0.25.2
 discord.py
 discord
-TensorFlow>=2.0
+torch