Added basic control flow

app.py

@@ -1,16 +1,21 @@
-import streamlit as st
+import streamlit
 from diffusers import StableDiffusionPipeline
 import torch
 
 model_id = "dream-textures/texture-diffusion"
 
-pipe = StableDiffusionPipeline.from_pretrained(model_id) #, torch_dtype=torch.float16)
-#pipe = pipe.to("cuda")
+if torch.cuda.is_available():
+    pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+    pipe = pipe.to("cuda")
+else:
+    pipe = StableDiffusionPipeline.from_pretrained(model_id)
 
-prompt = "pbr brick wall"
-image = pipe(prompt).images[0]  
-    
+examplePrompt = "pbr brick wall"
+streamlit.title("Texel Space Interpolation")
+prompt = streamlit.text_input(label="Prompt", value=examplePrompt)
+
+image = pipe(prompt).images[0]
 image.save("bricks.png")
 
-x = st.slider('Select a value')
-st.write(x, 'squared is', x * x)
+#x = streamlit.slider('Select a value')
+#streamlit.write(x, 'squared is', x * x)