Update app.py

app.py

@@ -1,32 +1,32 @@
 
-# import streamlit as st
-# import numpy as np
-# import cv2
-# import warnings
-# import os
+import streamlit as st
+import numpy as np
+import cv2
+import warnings
+import os
 
-# # Suppress warnings
-# warnings.filterwarnings("ignore", category=FutureWarning)
-# warnings.filterwarnings("ignore", category=UserWarning)
-
-# # Try importing TensorFlow
-# try:
-#     from tensorflow.keras.models import load_model
-#     from tensorflow.keras.preprocessing import image
-# except ImportError:
-#     st.error("Failed to import TensorFlow. Please make sure it's installed correctly.")
-
-# # Try importing PyTorch and Detectron2
-# try:
-#     import torch
-#     import detectron2
-# except ImportError:
-#     with st.spinner("Installing PyTorch and Detectron2..."):
-#         os.system("pip install torch torchvision")
-#         os.system("pip install 'git+https://github.com/facebookresearch/detectron2.git'")
-
-#     import torch
-#     import detectron2
+# Suppress warnings
+warnings.filterwarnings("ignore", category=FutureWarning)
+warnings.filterwarnings("ignore", category=UserWarning)
+
+# Try importing TensorFlow
+try:
+    from tensorflow.keras.models import load_model
+    from tensorflow.keras.preprocessing import image
+except ImportError:
+    st.error("Failed to import TensorFlow. Please make sure it's installed correctly.")
+
+# Try importing PyTorch and Detectron2
+try:
+    import torch
+    import detectron2
+except ImportError:
+    with st.spinner("Installing PyTorch and Detectron2..."):
+        os.system("pip install torch torchvision")
+        os.system("pip install 'git+https://github.com/facebookresearch/detectron2.git'")
+
+    import torch
+    import detectron2
 
 
 # import streamlit as st