Update app.py

app.py

@@ -8,8 +8,9 @@ tokenizer = GPT2Tokenizer.from_pretrained(model_name)
 
 # Function to generate keywords based on a prompt
 def generate_keywords(prompt):
-    # Encode input prompt
-    inputs = tokenizer.encode(prompt, return_tensors="pt")
+    # Encode input prompt with an instruction to give only keywords
+    prompt_with_instruction = prompt + " -- please provide only keywords, no extra text."
+    inputs = tokenizer.encode(prompt_with_instruction, return_tensors="pt")
 
     # Generate output from model
     outputs = model.generate(inputs, max_length=50, num_return_sequences=1, no_repeat_ngram_size=2, top_k=50, top_p=0.95)