from flask import Flask, request, jsonify | |
from moviepy.editor import * | |
from vid2sum_model import Vid2SumModel | |
app = Flask(__name__) | |
model = Vid2SumModel() | |
def predict(): | |
file = request.files['video'] | |
file.save('uploaded_video.mp4') # Save the uploaded video file | |
summary_text = model.predict('uploaded_video.mp4') # Call your model's predict method | |
return jsonify({'summary_text': summary_text}) | |
if __name__ == '__main__': | |
app.run() | |