Update app.py

app.py

@@ -39,6 +39,8 @@ def resize_image(image, max_size=1024):
     
     return image
 
+# ... existing imports and model loading code ...
+
 def process(image, progress=gr.Progress()):
     if image is None:
         return None, None
@@ -97,6 +99,8 @@ def process(image, progress=gr.Progress()):
         print(f"Error processing image: {str(e)}")
         return None, None
 
+# ... rest of the code remains the same ...
+
 # Gradio interface
 with gr.Blocks(css="""
     @import url('https://fonts.googleapis.com/css2?family=Orbitron:wght@400;500;700&display=swap');