Spaces:
Runtime error
Runtime error
import json | |
from flask import Flask, jsonify, request | |
from semantic_search import SemanticSearch | |
search = SemanticSearch('cuda') | |
search.test_search() | |
app = Flask(__name__) | |
app.config['JSON_AS_ASCII'] = False | |
def health(): | |
return jsonify({"status": "ok"}) | |
def search_route(): | |
data = request.get_json() | |
query = data.get('query', '') | |
top = data.get('top', 10) | |
titles, docs, scores = search.search(query, top) | |
result = [{'title': str(item1), 'text': str(item2), 'relevance': str(item3)} for item1, item2, item3 in zip(titles, docs, scores)] | |
return jsonify(result) | |
if __name__ == '__main__': | |
app.run(debug=False, host='0.0.0.0', port=7868) | |