bhlewis commited on
Commit
59e6012
·
verified ·
1 Parent(s): 4e81f35

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +36 -0
app.py ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from flask import Flask, request, jsonify
2
+ from gradio_client import Client
3
+ import os
4
+
5
+ app = Flask(__name__)
6
+
7
+ # Initialize the Gradio client
8
+ gradio_client = Client("bhlewis/BERT-for-Patents_Semantic-Patent-Finder-v2")
9
+
10
+ @app.route('/api/predict', methods=['POST'])
11
+ def predict():
12
+ try:
13
+ data = request.json
14
+ query = data.get('query', '')
15
+ top_k = data.get('top_k', 5)
16
+
17
+ print(f"Received query: {query}")
18
+ print(f"Top K: {top_k}")
19
+
20
+ result = gradio_client.predict(
21
+ query,
22
+ top_k,
23
+ api_name="/predict"
24
+ )
25
+
26
+ print(f"Result: {result}")
27
+
28
+ return jsonify({"result": result})
29
+
30
+ except Exception as e:
31
+ print(f"Error: {str(e)}")
32
+ return jsonify({"error": str(e)}), 500
33
+
34
+ if __name__ == "__main__":
35
+ port = int(os.environ.get("PORT", 7860))
36
+ app.run(host='0.0.0.0', port=port)