Update main.py

main.py

@@ -161,6 +161,7 @@ def get_streams():
     top_p = 0.95
     repetition_penalty = 1.0
 
+
     generate_kwargs = dict(
         temperature=temperature,
         max_new_tokens=max_new_tokens,
@@ -170,20 +171,11 @@ def get_streams():
         seed=42,
     )
     prompt = f""" prompt: 
-    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.
+    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
     """
     formatted_prompt = format_prompt(prompt)