Update main.py

main.py

@@ -26,49 +26,44 @@ def format_prompt(message):
 
 @app.route('/get_course', methods=['POST'])
 def recommend():
-    temperature=0.9
-    max_new_tokens=256
-    top_p=0.95
-    repetition_penalty=1.0
-    content = request.json
-    user_degree = content.get('degree')
-    user_stream = content.get('stream')
-    user_semester = content.get('semester')
-    temperature = float(temperature)
-    if temperature < 1e-2:
-        temperature = 1e-2
-    top_p = float(top_p)
-
-    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 recommendataion engine for course recommendation for student based on below details.
-
-    Degree: {user_degree}
-    Stream: {user_stream}
-    Current Semester: {user_semester}
-
-
-    Based on above details recommend the courses that realtes to above details 
-    Note: Output should bevalid json format in below format:
-    {{"course1:course_name,course2:course_name,course3:course_name,...}}
-    
-    """
-    formatted_prompt = format_prompt(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
-        yield output
-    return jsonify({"ans":output})
+    temperature = 0.9
+    max_new_tokens = 256
+    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)
+
+        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"})
 
 @app.route('/get_mentor', methods=['POST'])
 def mentor():