Update app.py

app.py

@@ -65,4 +65,38 @@ st.markdown("Create images from textual descriptions with Dall-E!")
 st.button("Edit Photo")
 
 
-## [C] Haltmann Earth Divison 
+## [C] Haltmann Earth Divison 
+
+
+import streamlit as st
+import numpy as np
+from PIL import Image, ImageDraw
+from inpainting import HaltmannInpainter
+
+ 
+st.set_option('deprecation.showfileUploaderEncoding', False)
+
+ 
+def main():
+
+    # Load the image and initialize the inpainter.    
+    image = st.file_uploader("Choose an image", type=["png", "jpg"])
+
+    if image is not None:        
+        inpainter = HaltmannInpainter()        
+
+        # Get the dimensions of the uploaded image.        
+        width, height = Image.open(image).size        
+
+        # Resize the uploaded image if it is too large for our model (> 512x512).        
+        if width > 512 or height > 512:            
+            max_dim = max(width, height)            
+
+            resize_factor = 512 / max_dim            
+
+            width, height = int(width * resize_factor), int(height * resize_factor)            
+
+            image = np.array(Image.open(image).resize((width, height)))                          else:            
+            image = np.array(Image.open(image))                    
+
+        # Inpaint!