Update app.py

app.py

@@ -28,17 +28,13 @@ def detect_image(model_choice, input_image=None, sample_image_choice=None):
     model = load_model(models[model_choice])
     if sample_image_choice:
         input_image = Image.open(sample_images[sample_image_choice])
-    if input_image:
-        return model.detect_image(input_image)
-    return None
+    return model.detect_image(input_image)
 
 def detect_video(model_choice, input_video=None, sample_video_choice=None):
     model = load_model(models[model_choice])
     if sample_video_choice:
         input_video = sample_videos[sample_video_choice]
-    if input_video:
-        return model.detect_video(input_video)
-    return None
+    return model.detect_video(input_video)
 
 app = gr.Blocks()
 
@@ -77,4 +73,4 @@ with app:
                     outputs=video_output
                 )
 
-app.launch(share=True)
+app.launch(share=True)