Update main.py

main.py

@@ -13,13 +13,9 @@ with open(file_path, "r") as file:
 def home():
     return jsonify({"message": "Welcome to the Recommendation API!"})
 
-def format_prompt(message, history):
-  prompt = "<s>"
-  for user_prompt, bot_response in history:
-    prompt += f"[INST] {user_prompt} [/INST]"
-    prompt += f" {bot_response}</s> "
-  prompt += f"[INST] {message} [/INST]"
-  return prompt
+def format_prompt(message):
+    return f"<s>[INST] {message} [/INST]</s>"
+
 
 @app.route('/get_course', methods=['POST'])
 def recommend():
@@ -27,7 +23,7 @@ def recommend():
     max_new_tokens = 256
     top_p = 0.95
     repetition_penalty = 1.0
-    history=history
+
 
     content = request.json
     user_degree = content.get('degree')
@@ -51,7 +47,7 @@ def recommend():
     Note: Output should be valid json format in below format:
     {{"course1:course_name, course2:course_name, course3:course_name,...}}
     """
-    formatted_prompt = format_prompt(prompt,history)
+    formatted_prompt = format_prompt(prompt)
 
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=False, details=True, return_full_text=True)
     return jsonify({"ans": stream})
@@ -62,7 +58,6 @@ def mentor():
     max_new_tokens = 256
     top_p = 0.95
     repetition_penalty = 1.0
-    history=history
 
     content = request.json
     user_degree = content.get('degree')
@@ -94,7 +89,7 @@ def mentor():
     Note: Output should be valid json format in below format:
     {{"mentor1:mentor_name,mentor2:mentor_name,mentor3:mentor_name,...}}
     """
-    formatted_prompt = format_prompt(prompt,history)
+    formatted_prompt = format_prompt(prompt)
 
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=False, details=True, return_full_text=True)
     return jsonify({"ans": stream})