Update app.py

app.py

@@ -11,6 +11,16 @@ image = st.text_input('Image URL', '')
 
 #image uploader
 # image = st.file_uploader(label = "Upload your image here",type=['png','jpg','jpeg'])
+def check_suffix(file='yolov5s.pt', suffix=('.pt',), msg=''):
+    # Check file(s) for acceptable suffix
+    if file and suffix:
+        if isinstance(suffix, str):
+            suffix = [suffix]
+        for f in file if isinstance(file, (list, tuple)) else [file]:
+            s = Path(f).suffix.lower()  # file suffix
+            if len(s):
+                assert s in suffix, f"{msg}{f} acceptable suffix is {suffix}"
+
 
 def check_file(file, suffix=''):
     # Search/download file (if necessary) and return path