Update main.py

main.py

@@ -198,6 +198,38 @@ def get_streams():
     return jsonify({"ans": stream})
 
 
+@app.route('/get_streams_degree', methods=['POST'])
+def get_streams():
+    data = request.get_json()
+    degree = data.get('degree')
+
+    temperature = 0.9
+    max_new_tokens = 256
+    top_p = 0.95
+    repetition_penalty = 1.0
+
+    generate_kwargs = dict(
+        temperature=temperature,
+        max_new_tokens=max_new_tokens,
+        top_p=top_p,
+        repetition_penalty=repetition_penalty,
+        do_sample=True,
+        seed=42,
+    )
+    prompt = f""" prompt: 
+    You need to act like as recommendation engine.
+    List 5 streams/branches based on the degree provided.
+    For example, if the degree is 'bachelors', list 5 streams related to bachelor's degree.
+    Note: Output should be list in below format:
+    [branch1, branch2, branch3,...]
+    Return only answer not prompt and unnecessary stuff, also dont add any special characters or punctuation marks
+    """
+    formatted_prompt = format_prompt(prompt)
+
+    stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=False, details=False, return_full_text=False)
+    return jsonify({"ans": stream})
+
+
 
 @app.route('/get_education_profiles', methods=['GET'])
 def get_education_profiles():