Spaces:
Paused
Paused
default dark theme streamlit
Browse files- .streamlit/config.toml +2 -0
- sirm2024_stablediff/main.py +0 -10
.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)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|