AI嘘日記は、OpenAIのGPT-4o miniを使用して、入力された写真に対してスタイル転送によるイラスト化を行い、その結果を元にAIが嘘の日記を生成するWebアプリです。日記を入力すると、AIがその内容に合わせたイラストを生成し、そのイラストと日記をPDFファイルにまとめて表示します。
img2imgの使用モデルは、CompVis/stable-diffusion-v1-4を使用しています。
text2imgの使用モデルは、Cstabilityai/sdxl-turboを使用しています。
- OpenAIのAPIキーを取得し、app.pyの
OPEN_AI_API_KEYに設定 ※GPT-4o miniを使用 - Appleシリコン(M1以降のパソコン)の場合は
pc="cudaの箇所をpc="mps"に変更 app.pyを実行してコンソールに表示されるURLにアクセスすると、AI嘘日記のWebアプリが起動- 画像をアップロードし、実行ボタンを押すと、AI嘘日記が生成 ※写真サイズは最大1MB
- 文章を入力すると、写真がAIによって作成されます
- しばらくすると、webページが更新されpdfファイルが表示
pip install flask
pip install torch torchvision torchaudio
pip install python-dotenv
pip install transformers
pip install openai
pip install docx2pdf
pip install python-docx
pip install diffusers