Update app.py

app.py

@@ -9,7 +9,8 @@ import tempfile
 import pyttsx3
 import os
 from datetime import datetime
-
+import torch
+from torchvision.models.mobilenetv2 import MobileNetV2
 # ========== TTS ENGINE ==========
 def speak(text):
     engine = pyttsx3.init()
@@ -25,11 +26,13 @@ if "PKR" not in currency_type:
     st.stop()
 
 # ========== LOAD MODEL ==========
-model = models.mobilenet_v2(pretrained=False)
-model.classifier[1] = torch.nn.Linear(model.classifier[1].in_features, 2)
-model.load_state_dict(torch.load("pkr_currency_classifier.pt", map_location=torch.device('cpu')))
+
+torch.serialization.add_safe_globals({'MobileNetV2': MobileNetV2})
+
+model = torch.load("pkr_currency_classifier.pt", map_location='cpu', weights_only=False)
 model.eval()
 
+
 # ========== TRANSFORMS ==========
 transform = transforms.Compose([
     transforms.Resize((224, 224)),