Update app.py

app.py

@@ -44,7 +44,7 @@ class PromptEnhancerV2:
         self.model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
             models_root_path,
             torch_dtype=dtype,
-            # attn_implementation="flash_attention_2",
+            attn_implementation="flash_attention_2",
             device_map=device_map,
         )
         self.processor = AutoProcessor.from_pretrained(models_root_path)