Swapnil-101 commited on
Commit
fdce613
·
verified ·
1 Parent(s): e1c567a

new api route added

Browse files
Files changed (1) hide show
  1. main.py +32 -0
main.py CHANGED
@@ -187,6 +187,38 @@ def get_streams():
187
  stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=False, details=False, return_full_text=False)
188
  return jsonify({"ans": stream})
189
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
190
 
191
  if __name__ == '__main__':
192
  app.run(debug=True)
 
187
  stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=False, details=False, return_full_text=False)
188
  return jsonify({"ans": stream})
189
 
190
+
191
+
192
+ @app.route('/get_education_profiles', methods=['GET'])
193
+ def get_education_profiles():
194
+ temperature = 0.9
195
+ max_new_tokens = 256
196
+ top_p = 0.95
197
+ repetition_penalty = 1.0
198
+
199
+ generate_kwargs = dict(
200
+ temperature=temperature,
201
+ max_new_tokens=max_new_tokens,
202
+ top_p=top_p,
203
+ repetition_penalty=repetition_penalty,
204
+ do_sample=True,
205
+ seed=42,
206
+ )
207
+
208
+ sectors = ["engineering", "medical", "arts", "commerce", "science", "management"] # Example sectors
209
+ prompt = f"""prompt:
210
+ You need to act like a recommendation engine.
211
+ List all education-related profiles in sectors like {', '.join(sectors)}.
212
+ Note: Output should be a list in the below format:
213
+ [profile1, profile2, profile3,...]
214
+ Return only the answer, not the prompt or unnecessary stuff, and don't add any special characters or punctuation marks.
215
+ """
216
+
217
+ formatted_prompt = format_prompt(prompt)
218
+
219
+ education_profiles = client.text_generation(formatted_prompt, **generate_kwargs, stream=False, details=False, return_full_text=False)
220
+ return jsonify({"ans": education_profiles})
221
+
222
 
223
  if __name__ == '__main__':
224
  app.run(debug=True)