digimap-web-app / app.py
joshangngoching's picture
Create app.py
8b04a03 verified
raw
history blame
167 Bytes
import streamlit as st
from transformers import pipeline
semantic_segmentation = pipeline("image-segmentation", "nvidia/segformer-b1-finetuned-cityscapes-1024-1024")