Hugging Face's logo

Spaces:
SeoJunn
/
segmentation3
Sleeping

App Files Community
SeoJunn commited on
Commit
a87dded
·
1 Parent(s): fcdbac9

Refactor : delete Device

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -9,8 +9,6 @@ import torch
9
  import tensorflow as tf
10
  from PIL import ImageDraw
11
 
12
- device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
13
-
14
  # image segmentation 모델
15
  feature_extractor = SegformerFeatureExtractor.from_pretrained(
16
  "nvidia/segformer-b1-finetuned-cityscapes-1024-1024"
 
9
  import tensorflow as tf
10
  from PIL import ImageDraw
11
 
 
 
12
  # image segmentation 모델
13
  feature_extractor = SegformerFeatureExtractor.from_pretrained(
14
  "nvidia/segformer-b1-finetuned-cityscapes-1024-1024"