Hugging Face's logo

Spaces:
dakkoong
/
homework3
Sleeping

App Files Community
dakkoong commited on
Commit
78adffd
·
1 Parent(s): 3fe6fac

Update app.py

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