Update app.py

app.py

@@ -64,7 +64,7 @@ def get_original_image(url_input):
         image = Image.open(requests.get(url_input, stream=True).raw)
         return image
 
-def detect_objects(model_name, url_input, image_input, webcam_input, threshold):
+def detect_objects(model_name, url_input, image_input,threshold):
     # Extract model and feature extractor
     feature_extractor = AutoFeatureExtractor.from_pretrained(model_name)
     
@@ -77,8 +77,6 @@ def detect_objects(model_name, url_input, image_input, webcam_input, threshold):
         image = get_original_image(url_input)
     elif image_input is not None:
         image = image_input
-    elif webcam_input is not None:
-        image = webcam_input
     
     # Make prediction
     processed_outputs = make_prediction(image, feature_extractor, model)
@@ -103,9 +101,6 @@ models = ["nickmuchi/yolos-small-finetuned-license-plate-detection", "nickmuchi/
 urls = ["https://drive.google.com/uc?id=1j9VZQ4NDS4gsubFf3m2qQoTMWLk552bQ", "https://drive.google.com/uc?id=1p9wJIqRz3W50e2f_A0D8ftla8hoXz4T5"]
 images = [[path.as_posix()] for path in sorted(pathlib.Path('images').rglob('*.j*g'))]
 
-twitter_link = """
-[![](https://img.shields.io/twitter/follow/nickmuchi?label=@nickmuchi&style=social)](https://twitter.com/nickmuchi)
-"""
 
 css = '''
 h1#title {
@@ -150,6 +145,6 @@ with demo:
     url_but.click(fn=detect_objects, inputs=[options, url_input, img_input, slider_input], outputs=[img_output_from_url], queue=True)
     img_but.click(fn=detect_objects, inputs=[options, url_input, img_input, slider_input], outputs=[img_output_from_upload], queue=True)
 
-    gr.Markdown("![visitor badge](https://visitor-badge.glitch.me/badge?page_id=nickmuchi-license-plate-detection-with-yolos)")
+    #gr.Markdown("![visitor badge](https://visitor-badge.glitch.me/badge?page_id=nickmuchi-license-plate-detection-with-yolos)")
 
 demo.launch(debug=True)