vishalkatheriya commited on
Commit
5f4808b
·
verified ·
1 Parent(s): b31de6b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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-eurosat"
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