Update app.py

app.py

@@ -341,39 +341,39 @@
 # ///////////////////////////////////Working
 
 
-# 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 streamlit as st
+import numpy as np
+import cv2
+import warnings
+import os
 
-#     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
 
-# from detectron2.engine import DefaultPredictor
-# from detectron2.config import get_cfg
-# from detectron2.utils.visualizer import Visualizer
-# from detectron2.data import MetadataCatalog
+from detectron2.engine import DefaultPredictor
+from detectron2.config import get_cfg
+from detectron2.utils.visualizer import Visualizer
+from detectron2.data import MetadataCatalog
 
 # # Load the trained models
 # @st.cache_resource