Spaces:
Sleeping
Sleeping
Upload 3 files
Browse files- Dockerfile +8 -0
- app.py +6 -1
- requirements.txt +1 -1
Dockerfile
CHANGED
@@ -2,9 +2,17 @@ FROM python:3.10-slim
|
|
2 |
|
3 |
WORKDIR /app
|
4 |
|
|
|
|
|
|
|
|
|
5 |
RUN apt-get update && apt-get install -y --no-install-recommends gcc && \
|
6 |
apt-get clean && rm -rf /var/lib/apt/lists/*
|
7 |
|
|
|
|
|
|
|
|
|
8 |
COPY requirements.txt .
|
9 |
RUN pip install --no-cache-dir -r requirements.txt
|
10 |
|
|
|
2 |
|
3 |
WORKDIR /app
|
4 |
|
5 |
+
# Önbellek dizini oluştur
|
6 |
+
RUN mkdir -p /app/cache && chmod -R 777 /app/cache
|
7 |
+
|
8 |
+
# Sistem bağımlılıkları
|
9 |
RUN apt-get update && apt-get install -y --no-install-recommends gcc && \
|
10 |
apt-get clean && rm -rf /var/lib/apt/lists/*
|
11 |
|
12 |
+
# Ortam değişkenleri
|
13 |
+
ENV TRANSFORMERS_CACHE=/app/cache
|
14 |
+
ENV HF_HOME=/app/cache
|
15 |
+
|
16 |
COPY requirements.txt .
|
17 |
RUN pip install --no-cache-dir -r requirements.txt
|
18 |
|
app.py
CHANGED
@@ -1,3 +1,4 @@
|
|
|
|
1 |
import os
|
2 |
from flask import Flask, request, jsonify, render_template
|
3 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
@@ -9,6 +10,10 @@ app = Flask(__name__)
|
|
9 |
logging.basicConfig(level=logging.INFO)
|
10 |
logger = logging.getLogger(__name__)
|
11 |
|
|
|
|
|
|
|
|
|
12 |
# Daha küçük Türkçe model
|
13 |
MODEL_NAME = "savasy/turkish-gpt2"
|
14 |
|
@@ -31,7 +36,7 @@ def load_model():
|
|
31 |
|
32 |
@app.route('/')
|
33 |
def home():
|
34 |
-
return render_template('
|
35 |
|
36 |
@app.route('/generate', methods=['POST'])
|
37 |
def generate():
|
|
|
1 |
+
from functools import lru_cache # Bu satır eklendi
|
2 |
import os
|
3 |
from flask import Flask, request, jsonify, render_template
|
4 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
|
10 |
logging.basicConfig(level=logging.INFO)
|
11 |
logger = logging.getLogger(__name__)
|
12 |
|
13 |
+
# Önbellek dizini ayarı
|
14 |
+
os.environ['TRANSFORMERS_CACHE'] = '/app/cache'
|
15 |
+
os.makedirs('/app/cache', exist_ok=True)
|
16 |
+
|
17 |
# Daha küçük Türkçe model
|
18 |
MODEL_NAME = "savasy/turkish-gpt2"
|
19 |
|
|
|
36 |
|
37 |
@app.route('/')
|
38 |
def home():
|
39 |
+
return render_template('index.html')
|
40 |
|
41 |
@app.route('/generate', methods=['POST'])
|
42 |
def generate():
|
requirements.txt
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
flask==2.0.3
|
2 |
-
werkzeug==2.0.3
|
3 |
transformers==4.30.2
|
4 |
torch==1.13.1 --extra-index-url https://download.pytorch.org/whl/cpu
|
5 |
sentencepiece==0.1.99
|
|
|
1 |
flask==2.0.3
|
2 |
+
werkzeug==2.0.3
|
3 |
transformers==4.30.2
|
4 |
torch==1.13.1 --extra-index-url https://download.pytorch.org/whl/cpu
|
5 |
sentencepiece==0.1.99
|