error handling V2

app06.py

@@ -1,17 +1,26 @@
-%%writefile app06_.py
-
 import streamlit as st
 from transformers import SegformerForSemanticSegmentation, SegformerImageProcessor
 from PIL import Image
 import numpy as np
 import torch
+import os
 
 # Load the model and processor
-model_path = "/home/user/app/defectdetection/model"
-model = SegformerForSemanticSegmentation.from_pretrained(model_path, local_files_only=True)
-preprocessor = SegformerImageProcessor.from_pretrained(model_path, local_files_only=True)
+model_dir = "C://Users//nazli//OneDrive//Desktop//PCB-Defect-Detection-Segmentation//model11//model//"
+model = SegformerForSemanticSegmentation.from_pretrained(model_dir)
+processor = SegformerImageProcessor.from_pretrained(model_dir)
 model.eval()
 
+
+try:
+    model = SegformerForSemanticSegmentation.from_pretrained(model_dir, local_files_only=True)
+    preprocessor = SegformerImageProcessor.from_pretrained(model_dir, local_files_only=True)
+except Exception as e:
+    existing_files = os.listdir(model_dir)
+    error_msg = f"Error: {e}\nFiles in directory: {existing_files}"
+    st.write(error_msg)
+
+
 st.title("PCB Defect Detection")
 
 # Upload image in Streamlit