diffusers flask streamlit或gradio 简洁可视化AIGC文生图页面_如何使用diffuser加载并使用文生图模型

参考：
https://python-bloggers.com/2022/12/stable-diffusion-application-with-streamlit/
https://github.com/LowinLi/stable-diffusion-streamlit

本项目使用生成模型为：PixArt
参考：https://huggingface.co/PixArt-alpha/PixArt-XL-2-1024-MS

项目结构

本项目很简洁，暂时每次只能返回一张图片；gpu资源T4 16g；测试下来基本也只能支持同时一个人使用

flask：作为后端，加载diffusers 扩散模型，提供对外生成图片的接口
streamlit：作为前端，用户访问使用的页面，根据用户填写prompt去请求flask 后端接口取回扩散模型生成的图片，并可视化
1
2

代码

1）flask 后端

运行：python flask_end.py

flask_end.py

from flask import Flask, Response, request
from PIL import Image
import torch
import io
from diffusers import PixArtAlphaPipeline


app 
1
2
3
4
5
6
7