error handling

app06.py

@@ -1,3 +1,4 @@
+%%writefile app06_.py
 
 import streamlit as st
 from transformers import SegformerForSemanticSegmentation, SegformerImageProcessor
@@ -6,12 +7,9 @@ import numpy as np
 import torch
 
 # Load the model and processor
-model_dir = "/home/user/app/defectdetection/model"
-model = SegformerForSemanticSegmentation.from_pretrained(model_path)
-preprocessor = SegformerImageProcessor.from_pretrained(model_path)
-
-model = SegformerForSemanticSegmentation.from_pretrained(model_dir)
-processor = SegformerImageProcessor.from_pretrained(model_dir)
+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.eval()
 
 st.title("PCB Defect Detection")