supertskone commited on
Commit
d0bf3bf
·
verified ·
1 Parent(s): 6ba791a

Delete run.py

Browse files
Files changed (1) hide show
  1. run.py +0 -43
run.py DELETED
@@ -1,43 +0,0 @@
1
- import os
2
-
3
- # Create the cache directory if it doesn't exist and set permissions
4
- hf_cache_dir = '/app/hf_cache'
5
- os.makedirs(hf_cache_dir, exist_ok=True)
6
-
7
- # Set the HF_HOME environment variable
8
- os.environ['HF_HOME'] = hf_cache_dir
9
-
10
- # Application code starts here
11
-
12
- import logging
13
- from flask import Flask, request, jsonify
14
- from app.search_engine import PromptSearchEngine
15
-
16
- app = Flask(__name__)
17
-
18
- # Disable parallelism for tokenizers
19
- os.environ['TOKENIZERS_PARALLELISM'] = 'false'
20
-
21
- # Configure logging
22
- logging.basicConfig(level=logging.INFO)
23
- logger = logging.getLogger(__name__)
24
-
25
- search_engine = PromptSearchEngine()
26
-
27
- @app.route('/search', methods=['POST'])
28
- def search():
29
- data = request.get_json()
30
- query = data.get('query')
31
- n = data.get('n', 5)
32
- use_pinecone = data.get('use_pinecone', True)
33
-
34
- logger.info(f"Received query: {query} with n: {n} and use_pinecone: {use_pinecone}")
35
- results = search_engine.most_similar(query, n, use_pinecone)
36
- formatted_results = [{'score': score, 'prompt': prompt} for score, prompt in results]
37
- logger.info(f"Returning results: {formatted_results}")
38
- return jsonify(formatted_results)
39
-
40
-
41
- if __name__ == '__main__':
42
- logger.info("Starting Flask server")
43
- app.run(debug=True)