uisikdag commited on
Commit
0d915b2
·
verified ·
1 Parent(s): fdd5263

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -10,9 +10,11 @@ model_path = 'weights/yolov8n.pt'
10
 
11
  model_dict={"vit":"uisikdag/weed_vit_balanced",
12
  "deit":"uisikdag/weed_deit_balanced",
 
 
13
  "convnext":"uisikdag/weeds_convnext_balanced",
14
- "resnet":"uisikdag/weed_resnet_balanced",
15
- "beit":"uisikdag/weed_beit_balanced"}
16
 
17
 
18
  # Setting page layout
 
10
 
11
  model_dict={"vit":"uisikdag/weed_vit_balanced",
12
  "deit":"uisikdag/weed_deit_balanced",
13
+ "swin":"uisikdag/weeds_swin_balanced",
14
+ "beit":"uisikdag/weed_beit_balanced",
15
  "convnext":"uisikdag/weeds_convnext_balanced",
16
+ "resnet":"uisikdag/weed_resnet_balanced"
17
+ }
18
 
19
 
20
  # Setting page layout