import gradio as gr | |
import torch | |
import random | |
import numpy as np | |
from transformers import MaskFormerFeatureExtractor, MaskFormerForInstanceSegmentation | |
demo = gr.Interface( | |
inputs=[gr.Image()], | |
outputs=inputs, | |
title="MaskFormer Demo" | |
) | |
demo.launch() |