initial ai service

Dockerfile

@@ -0,0 +1,17 @@
+FROM python:3.10-slim
+
+ENV PYTHONDONTWRITEBYTECODE=1
+ENV PYTHONUNBUFFERED=1
+
+WORKDIR /app
+
+RUN apt-get update && apt-get install -y git && rm -rf /var/lib/apt/lists/*
+
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY . .
+
+EXPOSE 7860
+
+CMD ["python", "app.py"]

requirements.txt

@@ -1,9 +1,9 @@
-flask==3.0.0
-flask-caching==2.1.0
-transformers==4.36.0
---extra-index-url https://download.pytorch.org/whl/cpu
-torch==2.9.1+cpu
-redis==5.0.1
-accelerate==0.25.0
-gunicorn==21.2.0
-python-dotenv==1.0.1
+
+flask
+flask-caching
+redis
+python-dotenv
+torch
+transformers
+sentencepiece
+accelerate

transliteration_api.py

@@ -640,4 +640,5 @@ if __name__ == "__main__":
     print("   POST /cache/clear")
     print("   POST /cache/warm")
 
-    app.run(host="0.0.0.0", port=5000, debug=False, threaded=True)
+    port = int(os.environ.get("PORT", 7860))
+    app.run(host="0.0.0.0", port=port, debug=False, threaded=True)