Update app.py
Browse files
app.py
CHANGED
|
@@ -6,6 +6,7 @@ from io import BytesIO
|
|
| 6 |
# import tensorflow as tf
|
| 7 |
import streamlit as st
|
| 8 |
from diffusers import StableDiffusionPipeline
|
|
|
|
| 9 |
import torch
|
| 10 |
from accelerate import Accelerator
|
| 11 |
#model_id = "CompVis/stable-diffusion-v1-4"
|
|
|
|
| 6 |
# import tensorflow as tf
|
| 7 |
import streamlit as st
|
| 8 |
from diffusers import StableDiffusionPipeline
|
| 9 |
+
from diffusers import DiffusionPipeline
|
| 10 |
import torch
|
| 11 |
from accelerate import Accelerator
|
| 12 |
#model_id = "CompVis/stable-diffusion-v1-4"
|