Update app.py

app.py

@@ -37,7 +37,7 @@ COT_TEMPLATE = \
 model = SAMR1ForConditionalGeneration_qwen2p5.from_pretrained(
     MODEL_PATH,
     torch_dtype=torch.bfloat16,
-    attn_implementation="flash_attention_2",
+    attn_implementation="eager",
     ignore_mismatched_sizes=True,
 ).to(DEVICE)
 processor = AutoProcessor.from_pretrained(MODEL_PATH)
@@ -65,7 +65,7 @@ def preprocess(image_path: str, instruction: str):
     pixel_mean = torch.Tensor([123.675, 116.28, 103.53]).view(-1, 1, 1)
     pixel_std = torch.Tensor([58.395, 57.12, 57.375]).view(-1, 1, 1)
 
-    question_template = COT_TEMPLATE
+    question_template = DEFAULT_TEMPLATE
     system_template = QWEN2_SYS
 
     image = Image.open(image_path).convert(mode="RGB")