Update app.py

app.py

@@ -2,11 +2,11 @@ import streamlit as st
 import numpy as np
 from PIL import Image
 from io import BytesIO
-import tensorflow as tf
+import keras
 
 st.title("Complete image using an AI model trained on Flickr images with the Pix2pix architecture.")
 image = st.file_uploader("Upload an image", type=["jpg", "png","jpeg"])
-model = tf.keras.models.load_model('complete_.keras')
+model = keras.models.load_model('complete_.keras')
 
 if image :
   button = st.button("Complete")