Update app.py

app.py

@@ -1,10 +1,14 @@
 import subprocess
 
-# Install Flask using pip
-try:
-    subprocess.check_call(['pip', 'install', 'Flask'])
-except subprocess.CalledProcessError as e:
-    print(f"Error installing Flask: {e}")
+# Function to install a Python 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_package('tensorflow')
 
 # Import necessary libraries after installation
 from flask import Flask, request, jsonify