azmisahin commited on
Commit
3d571b8
·
verified ·
1 Parent(s): 12f0941

Upload 3 files

Browse files
Files changed (3) hide show
  1. Dockerfile +8 -0
  2. app.py +6 -1
  3. 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('templates/index.html')
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 # Flask 2.0.x ile uyumlu sürüm
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