fix zero

app.py

@@ -17,19 +17,14 @@ args = parse_args()
 
 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)
+    if model is None:
+        from model_lib.modules import MoMA_main_modal
+        # model = MoMA_main_modal(args).to(device, dtype=torch.float16)
     
     # generated_image = model.generate_images(rgb, subject, prompt, strength=strength, seed=seed)
     generated_image = rgb