Update app.py

app.py

@@ -34,7 +34,7 @@ if raw_image != 'Select image':
 	with st.spinner('Loading Model...'):
 		feature_extractor = DPTFeatureExtractor.from_pretrained("Intel/dpt-large-ade")
 		device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-		model = DPTForSemanticSegmentation.from_pretrained("Intel/dpt-large-ade",ignore_mismatched_sizes=True,num_labels=len(id2label), id2label=id2label, label2id=label2id)
+		model = DPTForSemanticSegmentation.from_pretrained("Intel/dpt-large-ade")
 		model = model.to(device)
 		model.eval()