Update app.py

app.py

@@ -2,7 +2,7 @@ import os
 import random
 import logging
 from twitchio.ext import commands
-from transformers import AutoTokenizer, AutoConfig, AutoModelForCausalLM
+from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 
 # Set up logging
@@ -29,41 +29,118 @@ missing_vars = [var for var in required_vars if not globals().get(var)]
 if missing_vars:
     raise ValueError(f"Missing environment variables: {', '.join(missing_vars)}")
 
-# Replace with your actual model name
+# Initialize the Hugging Face tokenizer and model for DialoGPT
 model_name = "jacksonstrut/tinyllama-1.1B-chat"
-
-# Disable tokenizer parallelism
-os.environ["TOKENIZERS_PARALLELISM"] = "false"
-
-# Load the tokenizer with use_fast=False
-tokenizer = AutoTokenizer.from_pretrained(
-    model_name,
-    token=HUGGINGFACE_API_TOKEN,
-    use_fast=False
-)
-
-# Ensure pad_token is set
-if tokenizer.pad_token is None:
-    tokenizer.pad_token = tokenizer.eos_token
-
-config = AutoConfig.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(
-    model_name,
-    config=config,
-    token=HUGGINGFACE_API_TOKEN
-)
+tokenizer = AutoTokenizer.from_pretrained(model_name, token=HUGGINGFACE_API_TOKEN)
+model = AutoModelForCausalLM.from_pretrained(model_name, token=HUGGINGFACE_API_TOKEN)
 model.to('cpu')
 
 # List of house music hooks to drop randomly
 HOUSE_MUSIC_HOOKS = [
-    # ... your list of hooks ...
+    "Feeling the vibe, let's get hyped!",
+    "Rave or die!",
+    "Rave to the g,rave!",
+    "He just strutts!",
+    "Dancing is what to do!",
+    "Let's get riggity riggity wrecked!",
+    "Sleep is for the week, not the weekend!",
+    "It's hard out here for a pimp!",
+    "Something for your mind!",
+    "Something for your body!",
+    "Something for your soul!",
+    "Not everyone understands this thing they call house music!",
+    "It's a spiritual thing!",
+    "Talkin' about that thing they call house music!",
+    "Don't mean a thing if it ain't got that swing!",
+    "Can't stop. Won't stop!",
+    "Shots! Shots! Shots!",
+    "Jack to the sound of the underground!",
+    "Jack your body!",
+    "Asking ya..to BELIEVE..in the MIX!",
+    "BeaverBankHouseMafia in the house baby!",
+    "Dancing all night long, can't stop the beat!",
+    "Got my hands up high, feeling alive!",
+    "The music's got me moving, can't resist the groove!",
+    "Let's ride this wave, we're on top of the world!",
+    "Feeling the energy, let it consume us all!",
+    "We're in the zone, lost in the music!",
+    "This beat is fire, can't stop dancing!",
+    "The rhythm is pumping, let's keep it real!",
+    "Get ready to party, we're on a roll!",
+    "Let's make some noise, and show them how it's done!",
+    "We're living for the moment, feeling alive tonight!",
+    "The music's got us hyped, let's keep it real!",
+    "We're in the house, and we're here to stay!",
+    "Let's rock this place, and make it feel like home!",
+    "Got my hands up high, feeling alive tonight!",
+    "The beat is strong, and so are we!",
+    "We're on the dance floor, living our best lives!",
+    "Let's keep the energy high, and the vibes bright!",
+    "Feeling the music deep inside my soul!",
+    "We're on a roll, can't stop feeling alive!",
+    "The music's got us moving, let's keep on dancing!",
+    "Feeling the magic, in every single note!",
+    "We're lost in the rhythm, and we can't be stopped!",
+    "The music's got us swaying, back and forth!",
+    "Let's keep dancing all night long!",
+    "Feeling the heat, let it consume me whole!",
+    "We're in the zone, can't stop grooving!",
+    "The music's got us high, and we won't come down!",
+    "Let's keep on moving, until the morning light!",
+    "We're in this together, let's make it feel right!",
+    "The rhythm is strong, and we can't be stopped!",
+    "Feeling the beat, deep inside my soul!",
+    "Got my heart racing, and my feet are on fire!",
+    "The music's got us moving, let's keep it real!",
+    "Feeling the energy, and we won't stop until the break of dawn!",
+    "Got my hands up high, feeling alive tonight!",
+    "The beat is strong, and so are we!",
+    "Got my feet tapping, and my heart is pounding!",
+    "Feeling the magic, in every single note!",
+    "Let's keep on dancing all night long, feeling alive tonight!",
+    "We're lost in the rhythm, and can't be stopped!",
+    "Feeling the energy, let it consume me whole!",
+    "Got my heart racing, and my feet are on fire!",
+    "The music's got us moving, can't stop dancing!",
+    "Feeling the beat, deep inside my soul!",
+    "Let's keep on dancing, until we reach our goal!",
+    "We're on a roll, can't stop living our best lives!",
+    "The music's got us moving, can't stop swaying!",
+    "Feeling the energy, let it consume me whole!",
+    "We're in this together, let's make it feel right!",
+    "Got my heart racing, and my feet are on fire!",
+    "Feeling the magic, in every single note!",
+    "We're lost in the rhythm, and can't be stopped!",
+    "The music's got us swaying, back and forth!",
+    "Feeling the heat, let it consume me whole!",
+    "The beat is strong, and so are we!",
+    "Got my feet tapping, and my heart is pounding!",
+    "Let's keep dancing all night long, feeling alive tonight!",
+    "Feeling the energy, let it take control!",
+    "We're lost in the music, can't be found!",
+    "Got my hands up high, feeling alive tonight!",
+    "Feeling the vibe, let's keep it real!",
+    "We're on a roll, can't stop living our best lives!",
+    "Got my heart racing, and my feet are on fire!",
+    "The music's got us moving, can't stop grooving!",
+    "Feeling the magic, in every single note!",
+    "We're lost in the rhythm, and can't be stopped!",
+    "Got my feet tapping, and my heart is pounding!",
+    "Feeling the energy, let it take control!",
+    "We're on a roll, can't stop living our best lives!",
+    "The beat is strong, and so are we!",
+    "Feeling the energy, let it consume me whole!",
+    "Got my feet tapping, and my heart is pounding!",
+    "Feeling the beat, deep inside my soul!",
+    "Got my heart racing, and my feet are on fire!",
+    "The music's got us moving, can't stop dancing!",
 ]
 
 # Initialize chat history for users
 chat_histories = {}
 
 async def generate_response(user_id, user_message):
-    """Generates a response using the model."""
+    """Generates a response using the DialoGPT model."""
     try:
         # Retrieve or initialize the chat history for the user
         if user_id in chat_histories:
@@ -88,7 +165,7 @@ async def generate_response(user_id, user_message):
             do_sample=True,
             top_p=0.95,
             top_k=50,
-            pad_token_id=tokenizer.pad_token_id,
+            pad_token_id=tokenizer.eos_token_id,
             no_repeat_ngram_size=3,
         )
 
@@ -143,4 +220,4 @@ class TwitchChatBot(commands.Bot):
 # Initialize and run the bot
 if __name__ == "__main__":
     bot = TwitchChatBot()
-    bot.run()
+    bot.run()