hkhho commited on
Commit
d295684
·
1 Parent(s): 872be7d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -34,8 +34,9 @@ with col2:
34
  # Threshold ratio bar and class name input
35
  threshold_ratio = st.slider('Select threshold ratio:', min_value=0.0, max_value=1.0, step=0.1, value=0.6)
36
  class_name = st.text_input('Enter class name:', value='agumon')
37
-
38
- if uploaded_image is not None and uploaded_query is not None:
 
39
  # Process input and query image
40
  text_queries = [class_name]
41
  target_sizes = torch.Tensor([image.size[::-1]])
 
34
  # Threshold ratio bar and class name input
35
  threshold_ratio = st.slider('Select threshold ratio:', min_value=0.0, max_value=1.0, step=0.1, value=0.6)
36
  class_name = st.text_input('Enter class name:', value='agumon')
37
+ #
38
+ start_button = st.button('Start prediction')
39
+ if uploaded_image is not None and uploaded_query is not None and start_button:
40
  # Process input and query image
41
  text_queries = [class_name]
42
  target_sizes = torch.Tensor([image.size[::-1]])