Clean handler.py

handler.py

@@ -1,43 +1,9 @@
 import torch
 import base64
 import io
-import sys
 from typing import Dict, Any
 from PIL import Image
 
-# Create a fake cosmos_guardrail module BEFORE any diffusers import
-class FakeCosmosSafetyChecker:
-    def __init__(self):
-        pass
-    
-    def __call__(self, frames, **kwargs):
-        return frames
-    
-    def check_text_safety(self, text):
-        return True
-    
-    def check_video_safety(self, frames):
-        return frames
-
-# Inject fake module into sys.modules
-import types
-fake_guardrail = types.ModuleType("cosmos_guardrail")
-fake_guardrail.CosmosSafetyChecker = FakeCosmosSafetyChecker
-sys.modules["cosmos_guardrail"] = fake_guardrail
-
-# Now patch diffusers to think cosmos_guardrail is available
-import diffusers.utils.import_utils as import_utils
-original_is_available = getattr(import_utils, 'is_cosmos_guardrail_available', lambda: False)
-import_utils.is_cosmos_guardrail_available = lambda: True
-
-# Also patch at pipeline level if needed
-try:
-    import diffusers.pipelines.cosmos.pipeline_cosmos2_video2world as cosmos_pipeline
-    cosmos_pipeline.is_cosmos_guardrail_available = lambda: True
-    cosmos_pipeline.CosmosSafetyChecker = FakeCosmosSafetyChecker
-except:
-    pass
-
 
 class EndpointHandler:
     def __init__(self, path: str = ""):