Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -10,7 +10,7 @@ def download_models(model_id):
|
|
10 |
return f"./{model_id}"
|
11 |
|
12 |
box_annotator = sv.BoxAnnotator()
|
13 |
-
category_dict = {0: 'arborio', 1: '
|
14 |
|
15 |
|
16 |
@spaces.GPU(duration=200)
|
|
|
10 |
return f"./{model_id}"
|
11 |
|
12 |
box_annotator = sv.BoxAnnotator()
|
13 |
+
category_dict = {0: 'arborio', 1: 'basmati', 2: 'ipsala', 3: 'jasmine', 4: 'karacadag'}
|
14 |
|
15 |
|
16 |
@spaces.GPU(duration=200)
|