aelius commited on
Commit
5d8c6f3
·
1 Parent(s): c723304

default dark theme streamlit

Browse files
.streamlit/config.toml ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ [theme]
2
+ base="dark"
sirm2024_stablediff/main.py DELETED
@@ -1,10 +0,0 @@
1
- from diffusers import DiffusionPipeline
2
- import matplotlib.pyplot as plt
3
- import torch
4
-
5
- pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, variant="fp16")
6
- pipe.to("cuda")
7
-
8
- prompt = "brain magnetic resonance imaging, kandinsky, high resolution, photorealistic"
9
- image = pipe(prompt=prompt).images[0]
10
- plt.imshow(image)