.

Dockerfile

@@ -1,25 +1,17 @@
-# ベースイメージとしてCUDA対応のUbuntuを使用
-FROM nvidia/cuda:12.2.2-base-ubuntu22.04
+# ベースイメージとしてPython 3.9を使用
+FROM python:3.9-slim
 
 # 作業ディレクトリを設定
 WORKDIR /app
 
-# 必要なパッケージをインストール
-RUN apt-get update && apt-get install -y --no-install-recommends \
-    python3-pip \
-    python3-dev \
-    && apt-get clean \
-    && rm -rf /var/lib/apt/lists/*
-
-# Pythonのシンボリックリンクを作成
-RUN ln -s /usr/bin/python3 /usr/bin/python
-
 # 必要なPythonライブラリをインストールするための依存ファイルをコピー
 COPY requirements.txt /app/requirements.txt
 
 # 必要なPythonパッケージをインストール
 RUN pip install --no-cache-dir -r requirements.txt
 
+RUN nvidia-smi
+
 # アプリケーションコードをコンテナにコピー
 COPY . /app
 

app.py

@@ -9,10 +9,9 @@ from datetime import datetime
 from pytz import timezone
 from scripts.survey import handle_form_submission, handle_visit_choice, handle_proceed, localize, script, generate_image, send_feedback
 
-def initialize_models():
-    # 初期化
-    initialize(_use_local=False, use_gpu=True, use_dotenv=True)
-    init_model(use_local=False)
+# 初期化
+initialize(_use_local=False, use_gpu=True, use_dotenv=True)
+init_model(use_local=False)
 
 def process_image(input_image, mode, weight1=None, weight2=None):
     tokyo_time = datetime.now(timezone('Asia/Tokyo')).strftime("%Y-%m-%d %H:%M:%S")  # 日本時間のタイムスタンプ
@@ -211,5 +210,4 @@ with gr.Blocks() as demo:
     # JavaScriptの読み込み
     demo.load(js=script)
 
-initialize_models()
 demo.launch()

requirements.txt

@@ -22,5 +22,4 @@ python-dotenv
 peft==0.11.1
 flask_limiter==3.7.0
 gevent==24.2.1
-gevent-websocket==0.10.1
-firebase-admin
+gevent-websocket==0.10.1