#!flask/bin/python
from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
from flask import Flask
from flask_restful import Api


from resources.video import Video


model = MBartForConditionalGeneration.from_pretrained("model-pl2en")
tokenizer = MBart50TokenizerFast.from_pretrained("model-pl2en", src_lang="pl_PL")

app = Flask(__name__)
api = Api(app)

api.add_resource(Video, '/api/video', resource_class_kwargs={'model': model,
                                                             'tokenizer': tokenizer})

if __name__ == '__main__':
    app.run(debug=True, host='0.0.0.0')