Spaces:
Runtime error
Runtime error
autonomous019
commited on
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 |
|