Update app.py

app.py

@@ -12,12 +12,12 @@ def check_installation(package):
     else:
         return True
 
-def install_tensorflow():
+def install_tf_keras():
     try:
-        subprocess.check_call(['pip', 'install', 'tensorflow==2.0'])
+        subprocess.check_call(['pip', 'install', 'tf-keras'])
         return True
     except subprocess.CalledProcessError as e:
-        st.error(f"TensorFlow installation failed with error: {e}")
+        st.error(f"Failed to install tf-keras package: {e}")
         return False
 
 def summarize(result):
@@ -58,12 +58,12 @@ def get_transcript(video_link):
         st.write("Invalid YouTube video link")
 
 def main():
-    if not check_installation("tensorflow"):
-        st.write("TensorFlow 2.0 is not installed. Installing...")
-        if install_tensorflow():
-            st.success("TensorFlow 2.0 has been successfully installed.")
+    if not check_installation("tf_keras"):
+        st.write("tf-keras is not installed. Installing...")
+        if install_tf_keras():
+            st.success("tf-keras has been successfully installed.")
         else:
-            st.error("Failed to install TensorFlow 2.0. Please install it manually and retry.")
+            st.error("Failed to install tf-keras. Please install it manually and retry.")
             return
 
     import tensorflow as tf