EdenSw commited on
Commit
85e524b
·
1 Parent(s): 76f3f34

Create server.py

Browse files
Files changed (1) hide show
  1. server.py +16 -0
server.py ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from flask import Flask, request, jsonify
2
+ from moviepy.editor import *
3
+ from vid2sum_model import Vid2SumModel
4
+
5
+ app = Flask(__name__)
6
+ model = Vid2SumModel()
7
+
8
+ @app.route('/predict', methods=['POST'])
9
+ def predict():
10
+ file = request.files['video']
11
+ file.save('uploaded_video.mp4') # Save the uploaded video file
12
+ summary_text = model.predict('uploaded_video.mp4') # Call your model's predict method
13
+ return jsonify({'summary_text': summary_text})
14
+
15
+ if __name__ == '__main__':
16
+ app.run()