Update app.py
Browse files
app.py
CHANGED
@@ -99,7 +99,7 @@ Links to HuggingFace Models:
|
|
99 |
- [hustlv/yolos-small](https://huggingface.co/hustlv/yolos-small)
|
100 |
"""
|
101 |
|
102 |
-
models = ["nickmuchi/yolos-small-finetuned-masks"]
|
103 |
urls = ["https://drive.google.com/uc?id=1VwYLbGak5c-2P5qdvfWVOeg7DTDYPbro"]
|
104 |
|
105 |
twitter_link = """
|
|
|
99 |
- [hustlv/yolos-small](https://huggingface.co/hustlv/yolos-small)
|
100 |
"""
|
101 |
|
102 |
+
models = ["nickmuchi/yolos-small-finetuned-masks","nickmuchi/yolos-base-finetuned-masks"]
|
103 |
urls = ["https://drive.google.com/uc?id=1VwYLbGak5c-2P5qdvfWVOeg7DTDYPbro"]
|
104 |
|
105 |
twitter_link = """
|