Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
-
|
|
|
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]])
|