fixed uploading on load

app.py

@@ -21,5 +21,7 @@ prompt = st.text_input("Enter a prompt:", "batman hitting the griddy in gotham")
 image_bytes = query({
 	"inputs": prompt,
 })
-# You can access the image with PIL.Image for example
-image = Image.open(io.BytesIO(image_bytes))
+
+if(image_bytes):
+    image = Image.open(io.BytesIO(image_bytes))
+    st.image(image, caption="Image generated by the model", use_column_width=True)