autonomous019 commited on
Commit
b676b33
1 Parent(s): bbf3945

deepmind conv

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -23,9 +23,9 @@ feature_extractor = ViTFeatureExtractor()
23
  #feature_extractor = ViTFeatureExtractor.from_pretrained("google/vit-base-patch16-224")
24
 
25
  #the following gets called by classify_image()
26
- feature_extractor = PerceiverFeatureExtractor.from_pretrained("deepmind/vision-perceiver-con")
27
- model = PerceiverForImageClassificationConvProcessing.from_pretrained("deepmind/vision-perceiver-con")
28
- #google/vit-base-patch16-224, deepmind/vision-perceiver-con
29
  image_pipe = ImageClassificationPipeline(model=model, feature_extractor=feature_extractor)
30
 
31
 
 
23
  #feature_extractor = ViTFeatureExtractor.from_pretrained("google/vit-base-patch16-224")
24
 
25
  #the following gets called by classify_image()
26
+ feature_extractor = PerceiverFeatureExtractor.from_pretrained("deepmind/vision-perceiver-conv")
27
+ model = PerceiverForImageClassificationConvProcessing.from_pretrained("deepmind/vision-perceiver-conv")
28
+ #google/vit-base-patch16-224, deepmind/vision-perceiver-conv
29
  image_pipe = ImageClassificationPipeline(model=model, feature_extractor=feature_extractor)
30
 
31