m7mdal7aj commited on
Commit
fb7ac8d
1 Parent(s): a066b93

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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(your_detection_model) # Replace with your actual detection 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.")