ashwath-vaithina-ibm commited on
Commit
3ee4e81
·
verified ·
1 Parent(s): 2b1ac28

Delete test.py

Browse files
Files changed (1) hide show
  1. test.py +0 -18
test.py DELETED
@@ -1,18 +0,0 @@
1
- from control import recommendation_handler
2
-
3
- import numpy as np
4
-
5
- prompt_json, _ = recommendation_handler.populate_json()
6
- prompt = """
7
- Hello how are you?
8
- . """
9
-
10
- model_id='sentence-transformers/all-MiniLM-L6-v2'
11
-
12
- embedding_fn = recommendation_handler.get_embedding_func(inference='local', model_id=model_id)
13
-
14
-
15
- print(recommendation_handler.split_into_sentences(prompt))
16
- recommendation_json = recommendation_handler.recommend_prompt(prompt, prompt_json, embedding_fn=embedding_fn, get_xy=True)
17
-
18
- print(recommendation_json)