addin small entrypoint

app.py

@@ -1,22 +1,25 @@
 import os
 import whisper
-from flask import Flask, jsonify, render_template, request, send_file
+from flask import Flask, jsonify, request
 from werkzeug.utils import secure_filename
 
 STARTING_SIZE = 'small'
 UPLOAD_FOLDER = 'uploads'
 ALLOWED_EXTENSIONS = {'ogg', 'mp3', 'mp4', 'wav',
                       'flac', 'm4a', 'aac', 'wma', 'webm', 'opus'}
-current_size = STARTING_SIZE
+normal_size = STARTING_SIZE
+small_size = 'base'
 
 app = Flask(__name__)
 app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
 
-model = whisper.load_model(current_size)
-model_en = whisper.load_model(f"{current_size}.en")
+model = whisper.load_model(normal_size)
+model_en = whisper.load_model(f"{normal_size}.en")
+model_small = whisper.load_model(small_size)
+model_small_en = whisper.load_model(f"{small_size}.en")
 
 
-def inference(audio_file):
+def inference(audio_file, model=model, model_en=model_en):
     audio = whisper.load_audio(audio_file)
     audio = whisper.pad_or_trim(audio)
     mel = whisper.log_mel_spectrogram(audio).to(model.device)
@@ -62,5 +65,21 @@ def index():
     return "nothing yet to see here"
 
 
+@app.route("/small", methods=['POST'])
+def index():
+    if 'file' not in request.files:
+        return "no file sent!"
+    uploaded_file = request.files['file']
+    if uploaded_file.filename == '':
+        return "no file sent!"
+    if uploaded_file and allowed_file(uploaded_file.filename):
+        filename = secure_filename(uploaded_file.filename)
+        uploaded_file.save(os.path.join(
+            app.config['UPLOAD_FOLDER'], filename))
+        results = inference(os.path.join(
+            app.config['UPLOAD_FOLDER'], filename), model_small, model_small_en)
+        return jsonify({"results": results})
+
+
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)