fix zero

app.py

@@ -6,7 +6,7 @@ import numpy as np
 import torch
 from pytorch_lightning import seed_everything
 from model_lib.utils import parse_args
-
+from model_lib.modules import MoMA_main_modal
 os.environ["CUDA_VISIBLE_DEVICES"]="0"
 
 title = "MoMA"
@@ -15,22 +15,14 @@ device = torch.device('cuda')
 seed_everything(0)
 args = parse_args()
 
-model = []
+model = MoMA_main_modal(args).to(device, dtype=torch.float16)
 
 @spaces.GPU
 def inference(rgb, subject, prompt, strength, seed):
     global model
-    print(f"model length: {len(model)}")    
-    
     seed = int(seed) if seed else 0
     seed = seed if not seed == 0 else np.random.randint(0,1000)
-    
-    if len(model) == 0:
-        from model_lib.modules import MoMA_main_modal
-        model += [MoMA_main_modal(args).to(device, dtype=torch.float16)]
-    
-    generated_image = model[0].generate_images(rgb, subject, prompt, strength=strength, seed=seed)
-
+    generated_image = model.generate_images(rgb, subject, prompt, strength=strength, seed=seed)
     return generated_image
 
 gr.Interface(