Update app.py

app.py

@@ -1,13 +1,14 @@
 import subprocess
 
-# Function to install a Python package using pip
+# Function to install a package using pip
 def install_package(package):
     try:
         subprocess.check_call(['pip', 'install', package])
     except subprocess.CalledProcessError as e:
         print(f"Error installing {package}: {e}")
 
-# Install TensorFlow using the function
+# Install Flask and TensorFlow using pip
+install_package('Flask')
 install_package('tensorflow')
 
 # Import necessary libraries after installation