Update app.py

app.py

@@ -4,6 +4,18 @@ import torchvision.transforms as transforms
 from PIL import Image
 from resnet import SupCEResNet  # Ensure the correct import path
 
+import subprocess
+import sys
+
+# Ensure torch is installed before importing
+try:
+    import torch
+except ModuleNotFoundError:
+    print("🚨 Torch not found! Installing...")
+    subprocess.run([sys.executable, "-m", "pip", "install", "torch", "torchvision", "torchaudio"], check=True)
+    import torch  # Try again after installation
+
+
 # ✅ Define class labels (from Clothing1M)
 class_labels = [
     "T-Shirt", "Shirt", "Knitwear", "Chiffon", "Sweater", "Hoodie",