Spaces:

Nathanwit
/

Image-Captioning-DL

Runtime error

Nathanwit commited on Dec 11, 2023

Commit

5b16dfd

1 Parent(s): 0738976

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,11 +5,9 @@ from transformers import AutoTokenizer, ViTFeatureExtractor, VisionEncoderDecode
 device='cpu'
 encoder_checkpoint = "Thibalte/captionning_project"
-decoder_checkpoint = "Thibalte/captionning_project"
-model_checkpoint = "Thibalte/captionning_project"
 feature_extractor = ViTFeatureExtractor.from_pretrained(encoder_checkpoint)
-tokenizer = AutoTokenizer.from_pretrained(decoder_checkpoint)
-model = VisionEncoderDecoderModel.from_pretrained(model_checkpoint).to(device)
 def predict(image,max_length=24, num_beams=4):

 device='cpu'
 encoder_checkpoint = "Thibalte/captionning_project"
 feature_extractor = ViTFeatureExtractor.from_pretrained(encoder_checkpoint)
+tokenizer = AutoTokenizer.from_pretrained("Thibalte/captionning_project")
+model = AutoModel.from_pretrained("Thibalte/captionning_project")
 def predict(image,max_length=24, num_beams=4):