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