Update app.py

app.py

@@ -1,32 +1,27 @@
-import subprocess
-
-# Command to clone the repository
-clone_command = [
-    'git', 'clone', '--depth', '1', '-b', 'v2.17.0', 'https://github.com/tensorflow/models.git'
-]
-
-# Run the command
-subprocess.run(clone_command, check=True)
-
-# Verify the contents of the cloned repository
-import os
-print(f"Files in 'models' directory:", os.listdir('models'))
-
 import zipfile
 import sys
 import os
 
+# Paths to the ZIP file and the extraction directory
+zip_file_path = '/home/user/app/models.zip'
+extract_dir = '/home/user/app/models/'
+
+# Ensure the extraction directory exists (create it if it doesn't)
+if not os.path.exists(extract_dir):
+    os.makedirs(extract_dir)
+
+# Unzipping the file
+try:
+    with zipfile.ZipFile(zip_file_path, 'r') as zip_ref:
+        zip_ref.extractall(extract_dir)
+        print(f"Files in '{extract_dir}' directory:", os.listdir(extract_dir))
+except Exception as e:
+    print(f"An error occurred: {e}"
+
 # Check current directory and list files
 print("Current Directory:", os.getcwd())
 print("Files in Directory:", os.listdir())
 
-with zipfile.ZipFile(r'/home/user/app/models.zip', 'r') as zip_ref:
-    zip_ref.extractall(r'/home/user/app/models/')
-    print("file extracted")
-    print("Current Directory:", os.getcwd())
-    print("Files in Directory:", os.listdir())
-
-
 sys.path.append('models')
 import numpy as np
 import tensorflow as tf