Update main.py

main.py

@@ -161,7 +161,6 @@ def get_streams():
     top_p = 0.95
     repetition_penalty = 1.0
 
-
     generate_kwargs = dict(
         temperature=temperature,
         max_new_tokens=max_new_tokens,
@@ -171,11 +170,20 @@ def get_streams():
         seed=42,
     )
     prompt = f""" prompt: 
-    You need to act like as recommendation engine.
-    List all 40+ streams/branches in like computer science, chemical engineering, aerospace , etc
-    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
+    You need to act like as a recommendation engine.
+    List all 40+ streams/branches along with brief descriptions in fields like computer science, chemical engineering, aerospace, etc.
+    Note: Output should be in the following format:
+    [{
+        "branch": "Computer Science",
+        "description": "Computer science is the study of computers and computational systems...",
+    },
+    {
+        "branch": "Chemical Engineering",
+        "description": "Chemical engineering is a branch of engineering that uses principles of chemistry...",
+    },
+    ...
+    ]
+    Return only the answer without the prompt and unnecessary stuff, and don't add any special characters or punctuation marks.
     """
     formatted_prompt = format_prompt(prompt)
 
@@ -184,6 +192,7 @@ def get_streams():
 
 
 
+
 @app.route('/get_education_profiles', methods=['GET'])
 def get_education_profiles():
     temperature = 0.9