AI Headshot Generatorは、開発者やメーカーがAIアプリケーションを構築するための素晴らしいスタート地点を提供するオープンソースプロジェクトです。このプロジェクトは、Astria.aiによって提供されるAIモデルトレーニングと推論を利用して、わずか数分でプロフェッショナルなAIヘッドショットを生成します。コードをフォークし、変更を加えて、人気のあるAI SaaSアプリを構築するための発射台として利用できます。
このプロジェクトは、Astriaの新しいpacks APIを使用するための変更がマージされ、コード内でプロンプトをハードコーディングする必要がなくなり、異なるプロンプトのパックを提供し、新しいFluxモデルのファインチューニングに簡単に切り替えることができます。packs APIへの移行時には、Vercel環境にNEXT_PUBLIC_TUNE_TYPE=packsとPACK_QUERY_TYPE=bothを追加してください。
アプリは、AstriaによるAIモデルトレーニングと推論、Next.jsによるアプリとランディングページ、SupabaseによるDBと認証、Resend(オプション)によるヘッドショット準備完了時のユーザーへのメール送信、ShadcnとTailwind CSSによるスタイル、Vercelによるデプロイメント、Stripeによる課金で動作します。
ローカルで実行するには、Vercelテンプレートを使用してSupabase/VercelとGitHubリポジトリをセットアップし、新しく作成されたリポジトリをクローンして依存関係をインストールします。SupabaseダッシュボードでMagic Link認証を設定し、Astriaアカウントを作成して.env.localファイルにAPIキーを入力します。Vercel Blobを設定して画像アップロードを有効にし、Resendアカウントを作成してモデルトレーニング完了時にユーザーにメールを送信するオプションを設定します。Stripeを設定してクレジットベースでユーザーに課金することも可能です。
開発サーバーを起動し、ブラウザで
良い結果を得るためには、モデルに教えるための画像サンプルが重要です。顔のクローズアップを強制し、顔が中央に来るようにトリミングすることを検討してください。サンプルには1人だけが写っている画像を使用し、サングラスや帽子などのアクセサリーを避け、顔がはっきりと見えるようにしてください。複数の顔や繰り返しの被写体、複数の手足などが含まれる歪んだ結果が得られた場合は、アップロードされたサンプルが同じ1:1の高さ/幅のアスペクト比であることを確認し、サンプルに複数の人物が写っていないことを確認し、生成時に「double torso, totem pole」をネガティブプロンプトに追加し、生成時の寸法もサンプルと同じ1:1の高さ/幅比であることを確認してください。
Headshot AIは、AIアバター、アニメポートレート、ストーリーイラスト、ペットポートレート、プロダクトショット、フードフォトグラフィー、アイコン、スタイル一貫性のあるアセットなど、Astriaの他の多くのユースケースをサポートするように簡単に適応できます。
Headshot AIへの貢献を歓迎します。改善の提案や大幅な変更がある場合は、問題を開いてください!コードベースに貢献したい場合は、新しいブランチを作成し、devを指すプルリクエストを開いてください。
サポートとリソースについては、までメールでお問い合わせください。Headshot AIはMITライセンスの下でリリースされています。