Update app.py

app.py

@@ -13,10 +13,11 @@ model = st.sidebar.selectbox("Choose Model", ("prodia", "pollinations", "Dall-e
 
 prompt = st.sidebar.text_input("Prompt", "")
 
-if model == "prodia" or model == "pollinations":
-    try:
-        resp = Client.create_generation(model, prompt)
-        image = Image.open(BytesIO(resp))
-        st.image(image, caption="Generated Image")
-    except Exception as e:
-        st.error(str(e))
+if prompt:
+    if model == "prodia" or model == "pollinations":
+        try:
+            resp = Client.create_generation(model, prompt)
+            image = Image.open(BytesIO(resp))
+            st.image(image, caption="Generated Image")
+        except Exception as e:
+            st.error(str(e))