Deleted description

app.py

@@ -44,16 +44,6 @@ if 'reuse_image' not in st.session_state:
 def set_image(img):
     st.session_state.reuse_image = img
 
-st.title("AI Photo Object Removal")
-
-st.image(open("assets/demo.png", "rb").read())
-
-st.markdown(
-    """
-    So you want to remove an object in your photo? You don't need to learn photo editing skills.
-    **Just draw over the parts of the image you want to remove, then our AI will remove them.**
-    """
-)
 uploaded_file = st.file_uploader("Choose image", accept_multiple_files=False, type=["png", "jpg", "jpeg"])
 
 if uploaded_file is not None:

src/core.py

@@ -64,9 +64,10 @@ ENERGY_MASK_CONST = 100000.0              # large energy value for protective ma
 MASK_THRESHOLD = 10                       # minimum pixel intensity for binary mask
 USE_FORWARD_ENERGY = True                 # if True, use forward energy algorithm
 
-device = torch.device("cpu")
+device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+print(device)
 model_path = "./assets/big-lama.pt"
-model = torch.jit.load(model_path, map_location="cpu")
+model = torch.jit.load(model_path, map_location=device)
 model = model.to(device)
 model.eval()