Hugging Face's logo

Spaces:
hafez1082
/
skinwise-pytorch
Sleeping

App Files Community
hafez1082 commited on
Commit
1e67db5
·
verified ·
1 Parent(s): 642f9be

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ app = FastAPI()
11
 
12
  # Load model and processor once at startup
13
  repo_name = "Jayanth2002/dinov2-base-finetuned-SkinDisease"
14
- image_processor = AutoImageProcessor.from_pretrained(repo_name, use_fast=False)
15
  model = AutoModelForImageClassification.from_pretrained(repo_name)
16
 
17
  # Class names
 
11
 
12
  # Load model and processor once at startup
13
  repo_name = "Jayanth2002/dinov2-base-finetuned-SkinDisease"
14
+ image_processor = AutoImageProcessor.from_pretrained(repo_name, use_fast=True)
15
  model = AutoModelForImageClassification.from_pretrained(repo_name)
16
 
17
  # Class names