pratikshahp commited on
Commit
079e2da
1 Parent(s): 5f8adcf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ def input_image_setup(uploaded_file):
18
  st.set_page_config(page_title="Image Detection")
19
  st.header("Object Detection Application")
20
  #Select your model
21
- models = ["facebook/detr-resnet-50", "ciasimbaya/ObjectDetection", "hustvl/yolos-tiny"] # List of supported models
22
  model_name = st.selectbox("Select model", models)
23
  processor = AutoImageProcessor.from_pretrained(model_name)
24
  model = AutoModelForObjectDetection.from_pretrained(model_name)
 
18
  st.set_page_config(page_title="Image Detection")
19
  st.header("Object Detection Application")
20
  #Select your model
21
+ models = ["facebook/detr-resnet-50","ciasimbaya/ObjectDetection","hustvl/yolos-tiny","microsoft/table-transformer-detection","valentinafeve/yolos-fashionpedia"] # List of supported models
22
  model_name = st.selectbox("Select model", models)
23
  processor = AutoImageProcessor.from_pretrained(model_name)
24
  model = AutoModelForObjectDetection.from_pretrained(model_name)