Initial commit

app.py

@@ -1,3 +1,5 @@
+import os
+
 import streamlit as st
 
 import torch
@@ -11,6 +13,8 @@ st.title('Play with Stable-Diffusion v1-4')
 
 model_id = "CompVis/stable-diffusion-v1-4"
 device = "cuda"
+auth_token = os.environ.get("StableDiffusion") or True
+
 
 with st.spinner(
 	text='Loading...'
@@ -39,7 +43,6 @@ def infer(prompt, samples=2, steps=30, scale=7.5, seed=25):
 	images = []
 	for image in images_list["sample"]:
 		images.append(image)
-
 	return images
 
 
@@ -49,25 +52,24 @@ with st.form(key='new'):
 
 	col1, col2, col3 = st.columns(3)
 
-	with col1:
-		with st.expander(label='Expand parameters'):
-			n_samples = st.select_slider(
-				label='Num images',
-				options=range(1, 5),
-				value=1
-			)
-
-			steps = st.select_slider(
-				label='Steps',
-				options=range(1, 61),
-				value=40
-			)
-
-			scale = st.select_slider(
-				label='Guidance Scale',
-				options=range(1, 21),
-				value=7
-			)
+	with st.expander(label='Expand parameters'):
+		n_samples = col1.select_slider(
+			label='Num images',
+			options=range(1, 5),
+			value=1
+		)
+
+		steps = col2.select_slider(
+			label='Steps',
+			options=range(1, 61),
+			value=40
+		)
+
+		scale = col3.select_slider(
+			label='Guidance Scale',
+			options=range(1, 21),
+			value=7
+		)
 
 	st.form_submit_button()