Update app.py

app.py

@@ -51,7 +51,7 @@ class SiameseNetwork(nn.Module, PyTorchModelHubMixin):
 
 
 # pretrained model file
-model_file = 'best_signature_mobilenet.safetensors' #config.safetensor_file
+model_file = 'model.safetensors' #config.safetensor_file
 
 
 # Function to compute similarity
@@ -71,7 +71,7 @@ def visualize_heatmap(model, image):
 
 # Load the pre-trained model from safeetesor file
 def load_pipeline():
-    model_file = 'best_signature_mobilenet.safetensors' #config.safetensor_file
+    model_file = 'model.safetensors' #config.safetensor_file
         # model_id = 'tensorkelechi/signature_mobilenet'
     model = SiameseNetwork() # model class/skeleton