minor edit

app.py

@@ -7,5 +7,5 @@ style = st.selectbox('Style', ['Picasso', 'Van Gogh'], index=None)
 
 prompt_lst = [organ, modality, style]
 if None not in prompt_lst:
-    prompt = ','.join() 
+    prompt = ','.join(prompt_lst) 
     print(prompt)

pyproject.toml

@@ -8,6 +8,11 @@ readme = "README.md"
 [tool.poetry.dependencies]
 python = "^3.11"
 streamlit = "^1.35.0"
+transformers = "^4.41.2"
+accelerate = "^0.31.0"
+invisible-watermark = ">=0.2.0"
+diffusers = ">=0.19.0"
+matplotlib = "^3.9.0"
 
 
 [build-system]

sirm2024_stablediff/main.py

@@ -0,0 +1,10 @@
+from diffusers import DiffusionPipeline
+import matplotlib.pyplot as plt
+import torch
+
+pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, variant="fp16")
+pipe.to("cuda")
+
+prompt = "brain magnetic resonance imaging, kandinsky, high resolution, photorealistic"
+image = pipe(prompt=prompt).images[0]
+plt.imshow(image)