Update app.py

app.py

@@ -1,12 +1,12 @@
 import streamlit as st
-import keras
 import numpy as np
 from PIL import Image
 from io import BytesIO
+import tensorflow as tf
 
 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 = keras.models.load_model('/content/drive/MyDrive/complete_.keras')
+model = tf.keras.models.load_model('complete_.keras')
 
 if image :
   button = st.button("Complete")