Spaces:

abidlabs
/

image-classifier

Runtime error

abidlabs HF Staff commited on Nov 5, 2022

Commit

e835ae7

1 Parent(s): 5fb3af5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ import time
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-feature_extractor = ViTFeatureExtractor.from_pretrained('google/vit-base-patch16-224').to(device)
 model = ViTForImageClassification.from_pretrained('google/vit-base-patch16-224').to(device)
 def predict(image):

 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+feature_extractor = ViTFeatureExtractor.from_pretrained('google/vit-base-patch16-224')
 model = ViTForImageClassification.from_pretrained('google/vit-base-patch16-224').to(device)
 def predict(image):