Update app.py

app.py

@@ -2,7 +2,10 @@ import subprocess
 import importlib
 
 # List of required packages
-packages = ['cv2', 'numpy', 'tensorflow', 'gradio']
+packages = ['cv2', 'numpy', 'tensorflow', 'gr']
+
+# Mapping of package names to their PyPI names
+package_mapping = {'cv2': 'opencv-python-headless'}
 
 # Check and install packages if needed
 for package in packages:
@@ -10,12 +13,11 @@ for package in packages:
         importlib.import_module(package)
     except ImportError:
         print(f"{package} not found. Installing...")
-        install_command = ['pip', 'install', package]
         
-        if package == 'cv2':
-            # OpenCV requires a different install command
-            install_command.append('opencv-python-headless')
-
+        # Get the correct package name from the mapping
+        install_package = package_mapping.get(package, package)
+        
+        install_command = ['pip', 'install', install_package]
         subprocess.run(install_command, check=True)
 
 # Now that all packages are installed, import them
@@ -27,6 +29,9 @@ import gradio as gr
 # Your code using these packages can follow here
 
 
+# Your code using these packages can follow here
+
+
 def remove_background_deeplab(image_path):
     # Load the TensorFlow Lite model
     interpreter = tf.lite.Interpreter(model_path="2.tflite")