Update app.py

app.py

@@ -121,7 +121,7 @@ def initialize_models(args):
         nn.Dropout(0.3),
         nn.Linear(10240, 2)
     )
-    task_head.load_state_dict(torch.load(os.path.join(args.model_path, 'task_head_final.pt')))
+    # task_head.load_state_dict(torch.load(os.path.join(args.model_path, 'task_head_final.pt')))
     task_head.eval()
 
     processor = AutoProcessor.from_pretrained(
@@ -271,7 +271,7 @@ def chat_step(image1, image2, text, height, width, steps, guidance,
             seed = torch.Generator(device="cpu").seed()
         torch.manual_seed(seed)
         # Generate
-        if do_image:
+        if True:
             # image generation pipeline
             siglip_hs = None
             if state['siglip_processor'] and image_paths: