Add application file

app.py

@@ -0,0 +1,53 @@
+import gradio as gr
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
+# Load the fine-tuned model and tokenizer from Hugging Face
+model_name = "smgriffin/pop-lyrics-generator-v1"  # Replace with your Hugging Face repo name
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
+
+# Define the generation function
+def generate_lyrics(artist):
+    prompt = f"Artist: {artist}\nLyrics:"
+    input_ids = tokenizer.encode(prompt, return_tensors="pt")
+    output = model.generate(
+        input_ids,
+        max_length=150,  # Maximum number of tokens
+        temperature=0.9,  # Adjust randomness
+        top_k=50,  # Top-k sampling
+        top_p=0.95,  # Nucleus sampling
+        do_sample=True,  # Enable stochastic decoding
+    )
+    return tokenizer.decode(output[0], skip_special_tokens=True)
+
+# List of artists for the dropdown
+artists = [
+    "Madonna", "The Beach Boys", "Barbra Streisand", "Michael Jackson", "Mariah Carey", 
+    "Charli XCX", "Lana Del Rey", "Britney Spears", "Bjork", "Bee Gees", "Lady Gaga", 
+    "Kenny Rogers", "Rihanna", "Justin Bieber", "Dua Lipa", "Judy Garland", "Ed Sheeran", 
+    "Prince", "John Denver", "Elton John", "Meghan Trainor", "Katy Perry", "Demi Lovato", 
+    "Enrique Iglesias", "Miley Cyrus", "Joni Mitchell", "David Guetta", "Beyonce", 
+    "Avicii", "Backstreet Boys", "a-ha", "Sia", "Ellie Goulding", "Taylor Swift", 
+    "Maroon 5", "Billie Eilish", "U2", "Shawn Mendes", "Carly Rae Jepsen", "Lil Wayne", 
+    "Bob Dylan", "Sinad O'Connor", "Kenny Loggins", "Jimi Hendrix", "The Temptations", 
+    "Jason Derulo", "Shakira", "Sabrina Carpenter", "Rick Astley", "Owl City", 
+    "twenty one pilots", "Al Green", "Coldplay", "Bruno Mars", "Chris Brown", 
+    "Plain White T's", "5 Seconds of Summer", "John Lennon", "Olivia Rodrigo", 
+    "3OH!3", "Jon Bellion", "Lorde", "Tate McRae", "Snoop Dogg", "Mike Posner", 
+    "Selena Gomez & The Scene", "Drake", "Frank Ocean", "Tyler, The Creator", 
+    "Camila Cabello", "Pharrell Williams", "The Beatles", "Chappell Roan", 
+    "Kendrick Lamar", "SZA", "Post Malone", "Khalid", "Hozier", "The Weeknd", 
+    "Ariana Grande", "Doja Cat", "Gracie Abrams", "Bad Bunny", "Usher"
+]
+
+# Define the Gradio interface
+interface = gr.Interface(
+    fn=generate_lyrics,
+    inputs=gr.Dropdown(choices=artists, label="Select an Artist"),
+    outputs=gr.Textbox(label="Generated Lyrics"),
+    title="Lyrics Generator",
+    description="Select an artist from the dropdown and click 'Submit' to generate lyrics based on their style.",
+)
+
+# Launch the app
+interface.launch()

requirements.txt

@@ -0,0 +1,3 @@
+gradio
+transformers
+torch