Update main.py

main.py

@@ -198,41 +198,6 @@ def get_streams():
     return jsonify({"ans": stream})
     
 
-@app.route('/get_stream_degree', methods=['POST'])
-def get_stream_degree():
-    temperature = 0.9
-    max_new_tokens = 256
-    top_p = 0.95
-    repetition_penalty = 1.0
-
-    content = request.json
-    # user_degree = content.get('degree')  # Uncomment this line
-    user_stream = content.get('degree')
-
-    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 for streams/branches recommendation for a student. Below are current details.
-    Stream: {user_stream}
-    Based on current details recommend the streams/branches 
-    Note: Output should be list in below format:
-    [course1, course2, course3,...]
-    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():
     temperature = 0.9