Update app.py

app.py

@@ -50,6 +50,7 @@ def get_prompt(file:str):
         a=f.readlines()
     return a #a[0]:positive prompt, a[1] negative prompt
 
+@spaces.GPU
 def init_pipe():
     def unwarp_model(state_dict):
         new_state_dict = {}
@@ -124,7 +125,7 @@ def init_pipe():
     pipe.enable_model_cpu_offload()
 
     return pipe
-
+@spaces.GPU
 def inference(source_images, 
         target_images, 
         text_prompt, negative_prompt,