Spaces:
Runtime error
Runtime error
Commit
·
b676b33
1
Parent(s):
bbf3945
deepmind conv
Browse files
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-
|
| 27 |
-
model = PerceiverForImageClassificationConvProcessing.from_pretrained("deepmind/vision-perceiver-
|
| 28 |
-
#google/vit-base-patch16-224, deepmind/vision-perceiver-
|
| 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 |
|