coco-gelamay commited on
Commit
5828d1d
1 Parent(s): 2bd7a78

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -1,10 +1,10 @@
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
 
@@ -29,8 +29,8 @@ def object_classify(img1,img2):
29
  objects_2.append(j['label'])
30
 
31
  #gets the uncommon elements from the 2 lists
32
- missing_objects= *list(set(objects_1)-set(objects_2)), sep = ", "
33
-
34
  return missing_objects
35
 
36
 
 
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
 
 
29
  objects_2.append(j['label'])
30
 
31
  #gets the uncommon elements from the 2 lists
32
+ missing_objects= list(set(objects_1)-set(objects_2))
33
+ missing_objects= *missing_objects, sep = ", "
34
  return missing_objects
35
 
36