update app.py

README.md

@@ -7,7 +7,7 @@ sdk: gradio
 sdk_version: 3.43.2
 app_file: app.py
 pinned: false
-license: mit
+license: openrail
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -17,6 +17,7 @@ def convertColor(image, color):
 def process_image(image):
 	if image is None:
 		yield [None, None, None]
+		return
 
 	model = pipeline("image-segmentation")
 	scores = model(image)

requirements.txt

@@ -1,3 +1,4 @@
+gradio
 torch
 torchvision
 torchaudio