Spaces:
Runtime error
Runtime error
Commit
·
3c40940
1
Parent(s):
6cf4a47
Update app.py
Browse files
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")
|