Update main.py

main.py

@@ -167,11 +167,6 @@ def get_streams():
     repetition_penalty = 1.0
 
 
-    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,
@@ -182,10 +177,9 @@ def get_streams():
     )
     prompt = f""" prompt: 
     You need to act like as recommendation engine.
-    List all 40+ streams/branches in below degree 
-    Degree: {user_degree}
+    List all 40+ streams/branches in like computer science, chemical engineering, aerospace , etc
     Note: Output should be list in below format:
-    [course1, course2, course3,...]
+    [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)