Update app.py

app.py

@@ -1,19 +1,19 @@
 import streamlit as st
-import tensorflow as tf
+import keras
 import numpy as np
 from PIL import Image
 from io import BytesIO
 
 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('/content/drive/MyDrive/complete_.keras')
 
 if image :
   button = st.button("Complete")
   image = Image.open(image)
   image = image.convert("RGB")
   image = image.resize((128,128-80))
-  padded_width = image.width + padding_width  # Add the padding width to the original width
+  padded_width = image.width + 80 # Add the padding width to the original width
   padded_height =image.height  # Keep the original height
 
   padded_image = Image.new("RGB", (padded_width, padded_height), color="white")