Update app.py

app.py

@@ -29,6 +29,7 @@ if raw_image != 'Select image':
 
 	image = Image.open(raw_image)
 	image = np.asarray(image)
+	st.success("Load Image: Success")
 	
 	with st.spinner('Loading Model...'):
 		feature_extractor = DPTFeatureExtractor.from_pretrained("Intel/dpt-large-ade")
@@ -36,7 +37,8 @@ if raw_image != 'Select image':
 		model = DPTForSemanticSegmentation.from_pretrained("Intel/dpt-large-ade",ignore_mismatched_sizes=True,num_labels=len(id2label), id2label=id2label, label2id=label2id,reshape_last_stage=True)
 		model = model.to(device)
 		model.eval()
-
+		
+	st.success("Load model: Success")
 	with st.spinner('Preparing image...'):
 		# prepare the image for the model (aligned resize)
 		feature_extractor_inference = DPTFeatureExtractor(do_random_crop=False, do_pad=False)