Upload main.py

main.py

@@ -0,0 +1,30 @@
+from flask import Flask
+from flask_restx import Api, Resource, fields
+from werkzeug.datastructures import FileStorage
+import wav2vec2 as wave
+
+flask_app = Flask(__name__)
+app = Api(app = flask_app,
+          version="1.0",
+          title="Model deployment test",
+          description="Test your models and feel rad")
+
+upload_parser = app.parser()
+upload_parser.add_argument('file',
+                           location='files',
+                           type=FileStorage)
+
+
+model = app.model('Soundfile',
+		  {'name': fields.String(required = True,
+					 description="The soudnfile you want to decode",
+					 help="File is necessary")})
+
+@app.route('/transcribe', methods=['POST'])
+@app.expect(upload_parser)
+class Transcribe(Resource):
+    def post(self):
+        args = upload_parser.parse_args()
+        file = args.get('file')
+        text = wave.predict(file)
+        return text