update

Dockerfile

@@ -0,0 +1,22 @@
+FROM python:3.9
+
+WORKDIR /code
+
+COPY ./requirements.txt /code/requirements.txt
+
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+RUN apt-get -y update
+
+RUN useradd -m -u 1000 user
+
+USER user
+
+ENV HOME=/home/user \
+	PATH=/home/user/.local/bin:$PATH
+
+WORKDIR $HOME/app
+
+COPY --chown=user . $HOME/app
+
+CMD ["python", "main.py", "--host", "0.0.0.0", "--port", "7860"]

main.py

@@ -0,0 +1,25 @@
+from flask import Flask, request, jsonify
+from transformers import AutoProcessor, AutoModelForSpeechSeq2Seq
+from transformers import pipeline
+
+app = Flask(__name__)
+model = AutoModelForSpeechSeq2Seq.from_pretrained("GitNazarov/whisper-small-pt-3-uz")
+processor = AutoProcessor.from_pretrained("GitNazarov/whisper-small-pt-3-uz")
+
+@app.route('/', methods=['GET'])
+def index():
+    return jsonify({"message": "Welcome to whisper uz!"})
+
+@app.route('/transcribe', methods=['POST'])
+def transcribe():
+    data = request.json['data']
+    inputs = processor(data, return_tensors="pt", sampling_rate=16000, max_new_tokens=100)
+    input_features = inputs.input_features
+    generated_ids = model.generate(inputs=input_features)
+
+    transcription = processor.batch_decode(generated_ids, skip_special_tokens=True)
+    transcription = ''.join(transcription)
+    return str(transcription), {'Content-Type': 'application/json'}
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=7860)

requirements.txt

@@ -0,0 +1,2 @@
+flask
+transformers