Update app.py

app.py

@@ -16,7 +16,7 @@ tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)
 model = PeftModel.from_pretrained(model, peft_model_id)
 
 
-def make_inference(product_name, product_description):
+def make_inference(song_name):
     batch = tokenizer(
         f"This an Oasis song name, please write some lyrics.\n\n### song name:\n{song_name}: ",
         return_tensors="pt",