minor

app.py

@@ -150,12 +150,12 @@ if st.session_state.button_clicked:
 
         prompt = combined_prompts[style].format(image_description=image_description, organ_description=organ_description)
         print(prompt)
-    #     pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, variant="fp16")
-    #     pipe.to("cuda")
+        pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, variant="fp16")
+        pipe.to("cuda")
 
-    #     image = pipe(prompt=prompt).images[0]
+        image = pipe(prompt=prompt).images[0]
     
-    # st.image(image)
+    st.image(image)
     
     st.session_state.button_disabled = False