Spaces:

deep-learning-analytics
/

segformer_semantic_segmentation

Runtime error

deep-learning-analytics commited on Jan 4, 2022

Commit

e8c8cc8

1 Parent(s): 4626ab4

pillow import bug fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,6 +2,7 @@ import os
 import pandas as pd
 import numpy as np
 import torch
 from transformers import SegformerForSemanticSegmentation, SegformerFeatureExtractor
 from torch import nn
 import streamlit as st

 import pandas as pd
 import numpy as np
 import torch
+from PIL import Image
 from transformers import SegformerForSemanticSegmentation, SegformerFeatureExtractor
 from torch import nn
 import streamlit as st