Update src/app/model.py

src/app/model.py

@@ -12,19 +12,11 @@ from src.exception import CustomExceptionHandling
 
 
 # Install the required dependencies
-try:
-    subprocess.run(
-        "pip install -r requirements.txt",
-        env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
-        shell=True,
-        check=True,
-    )
-    logging.info("Dependencies installed successfully.")
-
-# Handle exceptions that may occur during the process
-except Exception as e:
-    # Custom exception handling
-    raise CustomExceptionHandling(e, sys) from e
+subprocess.run(
+    "pip install flash-attn --no-build-isolation",
+    env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
+    shell=True,
+)
 
 # Load model and processor from Hugging Face
 model_id = "microsoft/Florence-2-large-ft"