Update app.py
Browse files
app.py
CHANGED
@@ -34,7 +34,7 @@ def run_inference():
|
|
34 |
# Button to load KBVQA models
|
35 |
if st.button('Load KBVQA Models'):
|
36 |
# Call the function to load models and show progress
|
37 |
-
kbvqa = prepare_kbvqa_model(
|
38 |
|
39 |
if kbvqa:
|
40 |
st.write("Model is ready for inference.")
|
|
|
34 |
# Button to load KBVQA models
|
35 |
if st.button('Load KBVQA Models'):
|
36 |
# Call the function to load models and show progress
|
37 |
+
kbvqa = prepare_kbvqa_model('yolov5')
|
38 |
|
39 |
if kbvqa:
|
40 |
st.write("Model is ready for inference.")
|