Update app.py

app.py

@@ -1,20 +1,27 @@
 import zipfile
-import subprocess
 def unzip_content():
     try:
+        # First try using Python's zipfile
         print("Attempting to unzip content using Python...")
         with zipfile.ZipFile('./content.zip', 'r') as zip_ref:
             zip_ref.extractall('.')
     except Exception as e:
         print(f"Python unzip failed: {str(e)}")
         try:
+            # Fallback to system unzip command
             print("Attempting to unzip content using system command...")
             subprocess.run(['unzip', '-o', './content.zip'], check=True)
         except Exception as e:
             print(f"System unzip failed: {str(e)}")
             raise Exception("Failed to unzip content using both methods")
     print("Content successfully unzipped!")
-import subprocess
+
+# Try to unzip content at startup
+try:
+    unzip_content()
+except Exception as e:
+    print(f"Warning: Could not unzip content: {str(e)}")
+
 import gradio as gr
 import numpy as np
 import torch
@@ -28,18 +35,19 @@ from omegaconf import OmegaConf
 from CLIP import clip
 import os
 import sys
-import tempfile
-import io
-from pathlib import Path
-from PIL import Image
-import cv2
-import imageio
 
-# Add taming transformers path
+#os.chdir('./taming-transformers')
+#from taming.models.vqgan import VQModel
+#os.chdir('..')
 taming_path = os.path.join(os.getcwd(), 'taming-transformers')
 sys.path.append(taming_path)
 from taming.models.vqgan import VQModel
 
+
+from PIL import Image
+import cv2
+import imageio
+
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")