Update app.py

app.py

@@ -17,5 +17,8 @@ stability_api_key = st.text_input("Enter your Stability API key")
 replicate_api_token = st.text_input("Enter your Replicate API token")
 
 if st.button("Generate and Upscale Image"):
-    final_image_bytes = generate_and_upscale_image(text_prompt, clipdrop_api_key, stability_api_key, replicate_api_token)
-    st.markdown(get_image_download_link(final_image_bytes, "gfpgan_upscaled_image.png"), unsafe_allow_html=True)
+    final_image_bytes, error = generate_and_upscale_image(text_prompt, clipdrop_api_key, stability_api_key, replicate_api_token)
+    if error:
+        st.error(error)
+    else:
+        st.markdown(get_image_download_link(final_image_bytes, "gfpgan_upscaled_image.png"), unsafe_allow_html=True)