vishalkatheriya
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -6,7 +6,7 @@ import torch
|
|
6 |
from io import BytesIO
|
7 |
|
8 |
# Load the model and processor
|
9 |
-
repo_name = "vishalkatheriya18/convnextv2-tiny-1k-224-finetuned-
|
10 |
image_processor = AutoImageProcessor.from_pretrained(repo_name)
|
11 |
model = AutoModelForImageClassification.from_pretrained(repo_name)
|
12 |
|
|
|
6 |
from io import BytesIO
|
7 |
|
8 |
# Load the model and processor
|
9 |
+
repo_name = "vishalkatheriya18/convnextv2-tiny-1k-224-finetuned-bottomwear"
|
10 |
image_processor = AutoImageProcessor.from_pretrained(repo_name)
|
11 |
model = AutoModelForImageClassification.from_pretrained(repo_name)
|
12 |
|