Update src/streamlit_app.py

src/streamlit_app.py

@@ -5,7 +5,7 @@ import numpy as np
 import tempfile
 import torch
 import matplotlib.pyplot as plt
-from transformers import AutoFeatureExtractor, AutoModelForObjectDetection
+from transformers import AutoImageProcessor
 from PIL import Image, ImageDraw
 
 # Fix cache permission issue
@@ -15,7 +15,7 @@ model_id = "NaveenKumar5/Solar_panel_fault_detection"
 
 @st.cache_resource
 def load_model():
-    extractor = AutoFeatureExtractor.from_pretrained(model_id)
+    extractor = AutoImageProcessor.from_pretrained(model_id)
     model = AutoModelForObjectDetection.from_pretrained(model_id)
     return extractor, model