Update app.py

app.py

@@ -40,12 +40,25 @@ def check_file(file, suffix=''):
 st.title("Hololive Waifu Classification")
 
 image = st.text_input('Image URL', '')
+imgsz = st.text_input('Image Size', 1280)
+conf = st.text_input('Confidence threshold', 0.25)
+iou = st.text_input('IoU threshold', 0.45)
+multi_label = st.selectbox('Multiple labels per box', (False, True))
+agnostic = st.selectbox('Class-agnostic', (False, True))
+amp = st.selectbox('Automatic Mixed Precision inference', (False, True))
+max_det  = st.text_input('Maximum number of detections per image', 1000)
 
-if image != '':
+if st.button('Excute'):
     image = check_file(image)
     input_image = Image.open(image)
     model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt')
-    results = model(input_image, size=1280)
+    model.conf = float(conf)
+    model.max_det = int(max_det)
+    model.iou = float(iou)
+    model.agnostic = agnostic
+    model.multi_label = multi_label
+    model.amp = amp 
+    results = model(input_image, size=int(imgsz))
     for img in results.render():
         st.image(img)
-    st.write(results.pandas().xyxy[0])
+    st.write(results.pandas().xyxy[0])