Update app.py

app.py

@@ -1,7 +1,5 @@
-# app.py
-
 import streamlit as st
-from main import generate_image_from_text, upscale_image_stable_diffusion, further_upscale_image
+from main import generate_image_from_text, upscale_image_esrgan, further_upscale_image
 
 def main():
     st.title("Image Generation and Upscaling")
@@ -13,8 +11,8 @@ def main():
         st.success("Generating image from text prompt...")
         image_bytes = generate_image_from_text(prompt)
         
-        st.success("Upscaling image with stable-diffusion-x4-latent-upscaler...")
-        upscaled_image_bytes = upscale_image_stable_diffusion(image_bytes)
+        st.success("Upscaling image with ESRGAN...")
+        upscaled_image_bytes = upscale_image_esrgan(image_bytes)
         
         st.success("Further upscaling image with GFPGAN...")
         img = further_upscale_image(upscaled_image_bytes)
@@ -22,4 +20,4 @@ def main():
         st.image(img, caption='Upscaled Image', use_column_width=True)
 
 if __name__ == "__main__":
-    main()
+    main()