coco-gelamay commited on
Commit
3c40940
1 Parent(s): 6cf4a47

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -3
app.py CHANGED
@@ -1,11 +1,8 @@
1
  from transformers import SegformerFeatureExtractor, SegformerForSemanticSegmentation
2
- #from datasets import load_dataset
3
  from transformers import pipeline
4
  from PIL import Image, ImageDraw, ImageFont
5
  import gradio as gr
6
 
7
- #ds = load_dataset("hf-internal-testing/fixtures_ade20k", split="test")
8
-
9
  def object_classify(img1,img2):
10
 
11
  feature_extractor = SegformerFeatureExtractor.from_pretrained("nvidia/segformer-b2-finetuned-ade-512-512")
 
1
  from transformers import SegformerFeatureExtractor, SegformerForSemanticSegmentation
 
2
  from transformers import pipeline
3
  from PIL import Image, ImageDraw, ImageFont
4
  import gradio as gr
5
 
 
 
6
  def object_classify(img1,img2):
7
 
8
  feature_extractor = SegformerFeatureExtractor.from_pretrained("nvidia/segformer-b2-finetuned-ade-512-512")