Update generate.py

generate.py

@@ -1,7 +1,20 @@
 # generate.py
-# --- VERSION 10 (with Upscaling) ---
-
-print("--- RUNNING GENERATE.PY VERSION 10 (with Upscaling) ---")
+# --- VERSION 11 (with Monkey-Patch) ---
+
+print("--- RUNNING GENERATE.PY VERSION 11 (with Monkey-Patch) ---")
+
+# --- MONKEY-PATCH FOR OLD TORCHVISION ---
+# This block fixes the "ModuleNotFoundError" by tricking the system.
+# It tells any library looking for the new `functional_tensor` module
+# to use the old `functional` module instead.
+try:
+    import sys
+    import torchvision.transforms.functional as F
+    sys.modules['torchvision.transforms.functional_tensor'] = F
+    print("--- Successfully applied torchvision monkey-patch. ---")
+except Exception as e:
+    print(f"--- Could not apply torchvision monkey-patch: {e} ---")
+# --- END OF PATCH ---
 
 import torch
 import cv2