Spaces:
Sleeping
Sleeping
ailab
commited on
Commit
·
a412dbf
1
Parent(s):
8f2aa2e
segmentation
Browse files
app.py
CHANGED
|
@@ -6,7 +6,6 @@ from PIL import Image
|
|
| 6 |
import torch
|
| 7 |
from transformers import AutoImageProcessor, AutoModelForSemanticSegmentation
|
| 8 |
|
| 9 |
-
# ✅ PyTorch 모델로 변경
|
| 10 |
MODEL_ID = "nvidia/segformer-b5-finetuned-ade-640-640"
|
| 11 |
processor = AutoImageProcessor.from_pretrained(MODEL_ID)
|
| 12 |
model = AutoModelForSemanticSegmentation.from_pretrained(MODEL_ID)
|
|
@@ -103,14 +102,15 @@ def run_inference(input_img):
|
|
| 103 |
demo = gr.Interface(
|
| 104 |
fn=run_inference,
|
| 105 |
inputs=gr.Image(type="numpy", label="Input Image"),
|
| 106 |
-
outputs=
|
| 107 |
examples=[
|
| 108 |
"ADE_val_00000001.jpeg",
|
| 109 |
"ADE_val_00001159.jpg",
|
| 110 |
"ADE_val_00001248.jpg",
|
| 111 |
"ADE_val_00001472.jpg"
|
| 112 |
],
|
| 113 |
-
|
|
|
|
| 114 |
)
|
| 115 |
|
| 116 |
if __name__ == "__main__":
|
|
|
|
| 6 |
import torch
|
| 7 |
from transformers import AutoImageProcessor, AutoModelForSemanticSegmentation
|
| 8 |
|
|
|
|
| 9 |
MODEL_ID = "nvidia/segformer-b5-finetuned-ade-640-640"
|
| 10 |
processor = AutoImageProcessor.from_pretrained(MODEL_ID)
|
| 11 |
model = AutoModelForSemanticSegmentation.from_pretrained(MODEL_ID)
|
|
|
|
| 102 |
demo = gr.Interface(
|
| 103 |
fn=run_inference,
|
| 104 |
inputs=gr.Image(type="numpy", label="Input Image"),
|
| 105 |
+
outputs=gr.Plot(label="Overlay + Legend"),
|
| 106 |
examples=[
|
| 107 |
"ADE_val_00000001.jpeg",
|
| 108 |
"ADE_val_00001159.jpg",
|
| 109 |
"ADE_val_00001248.jpg",
|
| 110 |
"ADE_val_00001472.jpg"
|
| 111 |
],
|
| 112 |
+
flagging_mode="never",
|
| 113 |
+
cache_examples=False,
|
| 114 |
)
|
| 115 |
|
| 116 |
if __name__ == "__main__":
|