Update app.py

app.py

@@ -85,11 +85,11 @@ def load_pipeline():
     model = SiameseNetwork() # model class/skeleton
     
     # model.load_state_dict(torch.load(model_file))
-    model = load_model(model, model_file)
+    load_model(model, model_file)
     # model = pipeline('image-classification', model=model_id, device='cpu') 
     # model.eval()
     print(model)
-    return model[0] #.to('cpu')
+    return model #.to('cpu')
 
 # Streamlit app UI template
 st.title("Signature Forgery Detection")