Nepjune commited on
Commit
bf7d108
·
verified ·
1 Parent(s): 813ddfe

Upload API.py

Browse files
Files changed (1) hide show
  1. API.py +15 -0
API.py ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from flask import Flask, jsonify
2
+
3
+ app = Flask(__name__)
4
+
5
+ @app.route("/caption", methods=["POST"])
6
+ def caption():
7
+
8
+ image = request.files["image"]
9
+
10
+ caption = predict_caption([image])
11
+
12
+ return jsonify({"caption": caption})
13
+
14
+ if __name__ == "__main__":
15
+ app.run(debug=True)