supertskone commited on
Commit
0c62911
·
verified ·
1 Parent(s): 1907f70

Delete app/run.py

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