Update app.py

app.py

@@ -1,17 +1,21 @@
 import subprocess
 import sys
 
-# Ensure required packages are installed
+# Function to install a specific package version
 def install(package):
     subprocess.check_call([sys.executable, "-m", "pip", "install", package])
 
+# Ensure NumPy version is compatible
 try:
     import numpy as np
-    print(f"🤔NumPy version: {np.__version__}")
+    if not np.__version__.startswith("1."):
+        print(f"Detected NumPy version {np.__version__}. Downgrading to NumPy 1.x...")
+        install("numpy<2")
 except ImportError:
-    print("🤔NumPy is not installed.")
+    print("NumPy not found. Installing NumPy 1.x...")
     install("numpy<2")
 
+# Install other dependencies if necessary
 try:
     import torch
 except ImportError:
@@ -23,6 +27,14 @@ try:
 except ImportError:
     install("Pillow==9.5.0")
 
+# Import required libraries after ensuring installations
+import torch
+import torch.nn as nn
+import torch.nn.functional as F
+import torchvision.transforms as transforms
+from PIL import Image
+import gradio as gr
+
 # Import necessary libraries
 import torch
 import torch.nn as nn