|
import gradio as gr |
|
|
|
from transformers import SegformerFeatureExtractor, TFSegformerForSemanticSegmentation |
|
from PIL import Image |
|
import requests |
|
|
|
|
|
feature_extractor = SegformerFeatureExtractor.from_pretrained("nvidia/segformer-b0-finetuned-cityscapes-640-1280") |
|
model = TFSegformerForSemanticSegmentation.from_pretrained("nvidia/segformer-b0-finetuned-cityscapes-640-1280") |
|
|
|
|
|
url = "http://images.cocodataset.org/val2017/000000039769.jpg" |
|
image = Image.open(requests.get(url, stream=True).raw) |
|
|
|
|
|
|
|
inputs = feature_extractor(images=image, return_tensors="pt") |
|
outputs = model(**inputs) |
|
logits = outputs.logits |
|
|
|
|
|
|
|
def greet(): |
|
return outputs |
|
|
|
iface = gr.Interface( |
|
fn=sepia, |
|
inputs="text", |
|
outputs=["plot"]) |
|
iface.launch(share=True) |