fix naming stuff

ui/src/app/api/hf-jobs/route.ts

@@ -603,18 +603,20 @@ def generate_model_card_readme(repo_id: str, config: dict, model_name: str, cura
         is_instruction = lower_arch in instruction_arches or 'kontext' in lower_model_name
 
         datasets_config = config.get('config', {}).get('process', [{}])[0].get('datasets', [])
+        is_wan_video = lower_arch.startswith('wan2') or 'wan2' in lower_model_name
         is_video = (
             'video' in lower_arch
             or process_config.get('type') == 'video'
             or any(isinstance(dataset, dict) and dataset.get('do_i2v') for dataset in datasets_config)
-            or lower_arch.startswith('wan21')
-            or 'wan21' in lower_model_name
+            or is_wan_video
         )
 
-        if is_instruction:
+        if lower_arch.startswith('qwen_image_edit') or 'image-edit' in lower_model_name:
+            tags.append("image-to-image")
+        elif is_instruction:
             tags.append("image-to-image")
         elif is_video:
-            if 'i2v' in model_name.lower() or 'i2v' in lower_arch or 'i2v' in lower_model_name:
+            if is_wan_video and ('i2v' in model_name.lower() or 'i2v' in lower_arch or 'i2v' in lower_model_name):
                 tags.append("image-to-video")
             else:
                 tags.append("text-to-video")