Update anime_app.py

anime_app.py

@@ -11,10 +11,11 @@ from settings import (
 )
 from utils import randomize_seed_fn
 # if gr.NO_RELOAD:
-base_model = "nyxia/AAM-AnyLoRA-Anime-Mix"
-model = Model(base_model_id=base_model, task_name="NormalBae")
 
-# note: for high res 1024x1024, set num steps to 9 and guidance to 6
+@spaces.GPU
+def init():
+    base_model = "nyxia/AAM-AnyLoRA-Anime-Mix"
+    model = Model(base_model_id=base_model, task_name="NormalBae")
 
 def auto_process_image(image, prompt):
     a_prompt="anime style, cartoon, drawing, 2D anime, illustration, cartoon"
@@ -130,4 +131,5 @@ with gr.Blocks(theme=gr.themes.Soft(), css="footer {visibility: hidden}") as dem
 #    image.change(auto_process_image, inputs=[image, prompt], outputs=[result])
 
 if __name__ == "__main__":
+    init()
     demo.queue(max_size=1).launch()