Update app.py

app.py

@@ -7,6 +7,8 @@ import tempfile
 import os
 import asyncio
 from concurrent.futures import ThreadPoolExecutor
+import warnings
+from transformers.utils import logging
 
 # Set page configuration
 st.set_page_config(page_title="Solar Panel Fault Detection", layout="wide")
@@ -18,6 +20,10 @@ st.write("Upload a thermal video (MP4) of a solar panel to detect thermal, dust,
 # Load model and processor
 @st.cache_resource
 def load_model():
+    # Suppress warning about unused weights
+    warnings.filterwarnings("ignore", message="Some weights of the model checkpoint.*were not used")
+    logging.set_verbosity_error()
+    
     processor = DetrImageProcessor.from_pretrained("facebook/detr-resnet-50")
     model = DetrForObjectDetection.from_pretrained("facebook/detr-resnet-50")
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@@ -98,8 +104,7 @@ async def process_video(video_path, frame_skip=5, batch_size=4):
             
             while cap.isOpened():
                 ret, frame = cap.read()
-                if not ret:
-                    break
+                if not, break
                 
                 if frame_count % frame_skip != 0:
                     out.write(frame)