Update app.py

app.py

@@ -7,12 +7,11 @@ from modelscope.pipelines import pipeline
 from modelscope.utils.constant import Tasks
 from pathlib import Path
 
-# Initialize model with optimized settings
-@gr.on(app_started=True)
+# Initialize model
 def load_model():
     global img_colorization
     img_colorization = pipeline(
-        Tasks.image_colorization, 
+        Tasks.image_colorization,
         model='iic/cv_ddcolor_image-colorization',
         model_revision='v1.0.0'
     )
@@ -91,4 +90,5 @@ with gr.Blocks(theme="soft", title="🎨 AI Color Restoration Studio") as demo:
     clear_btn.add([input_img, output_img, status])
 
 if __name__ == "__main__":
-    demo.launch(debug=True)
+    load_model()
+    demo.launch(debug=True)