dakkoong commited on
Commit
aec538a
1 Parent(s): 78adffd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -7,6 +7,10 @@ from transformers import SegformerFeatureExtractor, TFSegformerForSemanticSegmen
7
  feature_extractor = SegformerFeatureExtractor.from_pretrained(
8
  "nvidia/segformer-b1-finetuned-cityscapes-1024-1024")
9
 
 
 
 
 
10
  # 모델 입력 크기를 확인합니다.
11
  input_size = model.input_shape[1:3]
12
 
 
7
  feature_extractor = SegformerFeatureExtractor.from_pretrained(
8
  "nvidia/segformer-b1-finetuned-cityscapes-1024-1024")
9
 
10
+ model = TFSegformerForSemanticSegmentation.from_pretrained(
11
+ "nvidia/segformer-b1-finetuned-cityscapes-1024-1024"
12
+ )
13
+
14
  # 모델 입력 크기를 확인합니다.
15
  input_size = model.input_shape[1:3]
16