Update app.py

app.py

@@ -63,7 +63,7 @@ def predict_image(img_path, keyword=""):
     return class_label, copywriting
 
 # Function to generate copywriting
-def generate_copywriting(class_label):
+def generate_copywriting(class_label,keyword):
     # Generate prompt based on class label
     prompt = f"{keyword.lower()}, {class_label.lower()}" if keyword else f"{class_label.lower()}"