coco-gelamay commited on
Commit
19696f5
·
1 Parent(s): 2db475a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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
 
@@ -39,7 +39,7 @@ DESCRIPTION = 'Input two indoor pictures. First image being the original and sec
39
  EXAMPLES = [['Bedroom_1.jpg'],['Bedroom_2.jpg']]
40
 
41
  INPUTS=[gr.inputs.Image(type = 'pil', label='Original Image'),gr.inputs.Image(type = 'pil', label='Second Image')]
42
- OUTPUTS=gr.outputs.Textbox("Missing Item/s is/are: ")
43
 
44
  interface=gr.Interface(object_classify,
45
  INPUTS,
 
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
 
 
39
  EXAMPLES = [['Bedroom_1.jpg'],['Bedroom_2.jpg']]
40
 
41
  INPUTS=[gr.inputs.Image(type = 'pil', label='Original Image'),gr.inputs.Image(type = 'pil', label='Second Image')]
42
+ OUTPUTS=gr.outputs.Textbox(label="Missing Item/s is/are: ")
43
 
44
  interface=gr.Interface(object_classify,
45
  INPUTS,