Update main.py

main.py

@@ -31,39 +31,36 @@ def recommend():
     top_p = 0.95
     repetition_penalty = 1.0
 
-    if request.method == 'POST':
-        content = request.json
-        user_degree = content.get('degree')
-        user_stream = content.get('stream')
-        user_semester = content.get('semester')
-
-        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 course recommendation for a student based on below details.
-        Degree: {user_degree}
-        Stream: {user_stream}
-        Current Semester: {user_semester}
-        Based on above details recommend the courses that relate to the above details 
-        Note: Output should be valid json format in below format:
-        {{"course1:course_name, course2:course_name, course3:course_name,...}}
-        """
-        formatted_prompt = format_prompt(prompt)
-        print(formatted_prompt)
-        stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
-        output = ""
-
-        for response in stream:
-            output += response.token.text
-        return jsonify({"ans": output})
-    else:
-        return jsonify({"error": "Invalid request method"})
+    content = request.json
+    user_degree = content.get('degree')
+    user_stream = content.get('stream')
+    user_semester = content.get('semester')
+
+    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 course recommendation for a student based on below details.
+    Degree: {user_degree}
+    Stream: {user_stream}
+    Current Semester: {user_semester}
+    Based on above details recommend the courses that relate to the above details 
+    Note: Output should be valid json format in below format:
+    {{"course1:course_name, course2:course_name, course3:course_name,...}}
+    """
+    formatted_prompt = format_prompt(prompt)
+    print(formatted_prompt)
+    stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
+    output = ""
+
+    for response in stream:
+        output += response.token.text
+    return jsonify({"ans": output})
 
 @app.route('/get_mentor', methods=['POST'])
 def mentor():