Spaces:

nttdataspain
/

Image-To-Text-Lora-ViT

Runtime error

D0k-tor commited on Jun 26, 2023

Commit

c8f3b13

1 Parent(s): 74dc664

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ os.environ['TF_ENABLE_ONEDNN_OPTS'] = '0'
 device='cpu'
-model_id = "nttdataspain/vit-gpt2-coco-lora"
 model = VisionEncoderDecoderModel.from_pretrained(model_id)
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 feature_extractor = ViTFeatureExtractor.from_pretrained(model_id)

 device='cpu'
+model_id = "nttdataspain/vit-gpt2-stablediffusion2-lora"
 model = VisionEncoderDecoderModel.from_pretrained(model_id)
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 feature_extractor = ViTFeatureExtractor.from_pretrained(model_id)