Update app.py

app.py

@@ -1,4 +1,3 @@
-%%file app.py
 import streamlit as st
 import numpy as np
 from PIL import Image
@@ -18,7 +17,7 @@ def load_model(model, model_path, device):
     return model
 
 # Load the model onto the specified device
-u2net = load_model(model=u2net, model_path="/content/u2net.pth", device="cpu")
+u2net = load_model(model=u2net, model_path="u2net.pth", device="cpu")
 
 # Mean and std for normalization
 mean = torch.tensor([0.485, 0.456, 0.406])