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)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|