Update app.py

app.py

@@ -4,18 +4,18 @@ import torch
 
 # Load the Phi 2 model and tokenizer
 tokenizer = AutoTokenizer.from_pretrained(
-    "arieridwans/phi_2-finetuned-lyrics",
+    "microsoft/phi-2",
     trust_remote_code=True
 )
 
 model = AutoModelForCausalLM.from_pretrained(
-    "microsoft/phi-2",
+    "arieridwans/phi_2-finetuned-lyrics",
     device_map="auto",
     trust_remote_code=True
 )
 
 # Streamlit UI
-st.title("Eleanor Rigby")
+st.title("Eleanor Rigby - Lyrics Generation")
 
 # User input prompt
 user_prompt = st.text_area("Enter your prompt that can be song lyrics:", """Yesterday, I saw you in my dream""")