Hugging Face's logo

Spaces:
shravvvv
/
SAG-ViT
Runtime error

App Files Community
shravvvv commited on
Commit
ae47ae6
·
1 Parent(s): 024824d

Updated app.py

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