Spaces:

flava
/

zero-shot-image-classification

Runtime error

aps commited on May 25, 2022

Commit

559ac78

1 Parent(s): ed3c0bf

Set eval mode

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,6 +7,7 @@ import torch
 images="dog.jpg"
 model = FlavaModel.from_pretrained("facebook/flava-full")
 fe = FlavaFeatureExtractor.from_pretrained("facebook/flava-full")
 tokenizer = BertTokenizer.from_pretrained("facebook/flava-full")

 images="dog.jpg"
 model = FlavaModel.from_pretrained("facebook/flava-full")
+model.eval()
 fe = FlavaFeatureExtractor.from_pretrained("facebook/flava-full")
 tokenizer = BertTokenizer.from_pretrained("facebook/flava-full")