Update main.py

main.py

@@ -13,8 +13,15 @@ with open(file_path, "r") as file:
 def home():
     return jsonify({"message": "Welcome to the Recommendation API!"})
 
+import random
+
 def format_prompt(message):
-    return f"<s>[INST] {message} [/INST]</s>"
+    # Generate a random user prompt and bot response pair
+    user_prompt = "UserPrompt"
+    bot_response = "BotResponse"
+
+    return f"<s>[INST] {user_prompt} [/INST] {bot_response}</s> [INST] {message} [/INST]"
+
 
 
 @app.route('/get_course', methods=['POST'])