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, description):
+def make_inference(travel_destination, description):
     batch = tokenizer(
         f"Below is a travel destination and description, please write a marketing email for this travel spot.\n\n### Travel Destination:\n{travel_destination}\n### Description:\n{description}\n\n### Marketing Email",
         return_tensors="pt",
@@ -35,10 +35,10 @@ if __name__ == "__main__":
     gr.Interface(
         make_inference,
         [
-            gr.inputs.Textbox(lines=2, label="Product Name"),
-            gr.inputs.Textbox(lines=5, label="Product Description"),
+            gr.inputs.Textbox(lines=2, label="Travel Destination"),
+            gr.inputs.Textbox(lines=5, label="One Line Description"),
         ],
         gr.outputs.Textbox(label="Ad"),
         title="TravelDestination-MarketMail-AI",
-        description="TravelDestination-MarketMail-AI is a tool that generates marketing emails for different popular travel sites.",
+        description="TravelDestination-MarketMail-AI is a tool that generates marketing emails based off one travel destination name and a short line of description",
     ).launch()