fix zero

app.py

@@ -15,13 +15,22 @@ device = torch.device('cuda')
 seed_everything(0)
 args = parse_args()
 
-from model_lib.modules import MoMA_main_modal
-model = MoMA_main_modal(args).to(device, dtype=torch.float16)
+model = None
+
+@spaces.GPU
+def load_model(model):
+    from model_lib.modules import MoMA_main_modal
+    model = MoMA_main_modal(args).to(device, dtype=torch.float16)
+    return model
 
 @spaces.GPU
 def inference(rgb, subject, prompt, strength, seed):
     seed = int(seed) if seed else 0
     seed = seed if not seed == 0 else np.random.randint(0,1000)
+    
+    if model is None:
+        model = load_model(model)
+    
     generated_image = model.generate_images(rgb, subject, prompt, strength=strength, seed=seed)
     return generated_image