commit

app.py

@@ -16,6 +16,9 @@ model = SegformerForSemanticSegmentation.from_pretrained(model_name)
 model.to(device)
 
 
+from datasets import load_dataset
+from PIL import Image
+
 if file_name is not None:
     image = Image.open(file_name)
     image.show()