Initial commit

app.py

@@ -1,16 +1,22 @@
 import whisper
 import numpy as np
 from datetime import datetime
+import torch
 from flask import Flask,request,jsonify
 from transformers import pipeline
 import os
 from dotenv import load_dotenv
+from flask_cors import CORS
+
 load_dotenv()
 app=Flask(__name__)
+CORS(app)
 model=whisper.load_model("base")
 SAMPLE_RATE=16000
 DURATION=10
 OUTPUT_FILE="recorded_audio.wav"
+
+device="cuda" if torch.cuda.is_available() else "cpu"
 translator = pipeline("translation", model="Helsinki-NLP/opus-mt-mul-en",token=os.environ.get("HUGGING_FACE_AUTH_TOKEN"))
 
 
@@ -101,5 +107,5 @@ def translation_endpoint():
         return jsonify({"status":"error","message":str(e)}),500
 
 if __name__=="__main__":
-    app.run(debug=True,host="0.0.0.0",port=5000)
+    app.run(debug=False,host="0.0.0.0",port=int(os.environ.get("PORT",7860)))
     

dockerfile

@@ -1,4 +1,6 @@
 FROM python:3.11-slim
+
+RUN apt-get update && apt-get install -y ffmpeg git && rm -rf /var/lib/apt/lists/*
 WORKDIR /app
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt