Update app.py

app.py

@@ -71,14 +71,16 @@ def visualize_heatmap(model, image):
 
 # Load the pre-trained model from safeetesor file
 def load_pipeline():
-    model_id = 'tensorkelechi/signature_mobilenet'
-    # model = SiameseNetwork() # model class/skeleton
+    model_file = 'best_signature_mobilenet.safetensors' #config.safetensor_file
+        # model_id = 'tensorkelechi/signature_mobilenet'
+    model = SiameseNetwork() # model class/skeleton
     
     # model.load_state_dict(torch.load(model_file))
-    model = pipeline('image-classification', model=model_id, device='cpu') 
+    model = load_model(model, model_file)
+    # model = pipeline('image-classification', model=model_id, device='cpu') 
     model.eval()
     
-    return model
+    return model.to('cpu')
 
 # Streamlit app UI template
 st.title("Signature Forgery Detection")