Hugging Face's logo

Spaces:
shravvvv
/
SAG-ViT
Runtime error

App Files Community
shravvvv commited on
Commit
696435b
·
1 Parent(s): ae47ae6

Updated app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from PIL import Image
4
  from transformers import AutoModelForImageClassification, AutoImageProcessor
5
 
6
 
7
- # Load model and processor with custom code enabled
8
  model = AutoModelForImageClassification.from_pretrained("shravvvv/SAG-ViT", trust_remote_code=True)
9
  processor = AutoImageProcessor.from_pretrained("shravvvv/SAG-ViT", trust_remote_code=True)
10
 
 
4
  from transformers import AutoModelForImageClassification, AutoImageProcessor
5
 
6
 
7
+ # Load model and processor with custom code enabled
8
  model = AutoModelForImageClassification.from_pretrained("shravvvv/SAG-ViT", trust_remote_code=True)
9
  processor = AutoImageProcessor.from_pretrained("shravvvv/SAG-ViT", trust_remote_code=True)
10