Update app.py

app.py

@@ -2,6 +2,7 @@ import os
 import subprocess
 from huggingface_hub import login
 
+
 def clone_and_initialize_repo(repo_url, target_dir):
     """
     Clone a GitHub repository using a personal access token and initialize submodules.
@@ -52,6 +53,7 @@ def huggingface_login():
         print(f"Error during Hugging Face login: {e}")
         raise
 
+
 def start_gradio_demo():
     """
     Start the Gradio demo.
@@ -64,6 +66,34 @@ def start_gradio_demo():
         raise
 
 
+def modify_file(file_path):
+    """
+    Modify specific lines in a file.
+    
+    Replace lines 107 and 108 with `raise ImportError`.
+    """
+    try:
+        # Read the file contents
+        with open(file_path, "r") as file:
+            lines = file.readlines()
+
+        # Modify the specified lines
+        lines[106] = "raise ImportError\n"  # Line 107 (index starts at 0)
+        lines[107] = ""  # Line 108, replace with an empty line or remove it
+
+        # Write the modified lines back to the file
+        with open(file_path, "w") as file:
+            file.writelines(lines)
+
+        print(f"File {file_path} modified successfully: lines 107 and 108 replaced.")
+    except FileNotFoundError:
+        print(f"Error: File {file_path} not found.")
+        raise
+    except Exception as e:
+        print(f"Error modifying file {file_path}: {e}")
+        raise
+
+
 def main():
     """
     Main function to perform all operations.
@@ -72,10 +102,16 @@ def main():
     repo_url = "https://github.com/dadwadw233/BoxDreamer.git"
     target_dir = "./BoxDreamer"
 
+    # Define the file to be modified
+    file_to_modify = "./BoxDreamer/three/dust3r/croco/models/pos_embed.py"
+
     try:
         # Clone the repository and initialize submodules
         clone_and_initialize_repo(repo_url, target_dir)
 
+        # Modify the specified file
+        modify_file(file_to_modify)
+
         # Log in to Hugging Face
         huggingface_login()