Update app.py

app.py

@@ -1,6 +1,9 @@
+# Add this code at the very beginning of your script, before any other imports
+
 import sys
 import types
 
+# Create a fake torch._six module with string_classes
 torch_six = types.ModuleType('torch._six')
 torch_six.string_classes = (str,)  # In Python 3, string_classes is just (str,)
 
@@ -12,6 +15,24 @@ if 'torch' not in sys.modules:
 sys.modules['torch._six'] = torch_six
 torch._six = torch_six
 
+# Monkey patch for pytorch_lightning.utilities.distributed
+try:
+    from pytorch_lightning.utilities.rank_zero import rank_zero_only
+    # Create the old module path
+    pl_utils_dist = types.ModuleType('pytorch_lightning.utilities.distributed')
+    pl_utils_dist.rank_zero_only = rank_zero_only
+    sys.modules['pytorch_lightning.utilities.distributed'] = pl_utils_dist
+except ImportError:
+    # If even the new import fails, create a dummy decorator
+    def rank_zero_only(fn):
+        """Dummy decorator that just returns the function as-is"""
+        return fn
+    
+    pl_utils_dist = types.ModuleType('pytorch_lightning.utilities.distributed')
+    pl_utils_dist.rank_zero_only = rank_zero_only
+    sys.modules['pytorch_lightning.utilities.distributed'] = pl_utils_dist
+
+# Now continue with your original imports
 from pydoc import describe
 import gradio as gr
 import torch
@@ -25,9 +46,9 @@ from ldm.util import instantiate_from_config
 from huggingface_hub import hf_hub_download
 import spaces
 
+# Rest of your code continues here...
 model_path_e = hf_hub_download(repo_id="multimodalart/compvis-latent-diffusion-text2img-large", filename="txt2img-f8-large.ckpt")
 
-#@title Import stuff
 import argparse, os, sys, glob
 import numpy as np
 from PIL import Image