ashwml commited on
Commit
8ce4fdb
·
1 Parent(s): f250863

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -46,12 +46,12 @@ import torch
46
 
47
  with open("model/config.json") as f:
48
  n=json.load(f)
49
- encoder._name_or_path=n["encoder"]["_name_or_path"]
50
- decoder._name_or_path=n["decoder"]["_name_or_path"]
51
 
52
 
53
- print(encoder._name_or_path,decoder._name_or_path,)
54
- model = VisionEncoderDecoderModel.from_encoder_decoder_pretrained(encoder._name_or_path,decoder._name_or_path)
55
 
56
 
57
  tokenizer = AutoTokenizer.from_pretrained(decoder._name_or_path)
 
46
 
47
  with open("model/config.json") as f:
48
  n=json.load(f)
49
+ encoder_name_or_path=n["encoder"]["_name_or_path"]
50
+ decoder_name_or_path=n["decoder"]["_name_or_path"]
51
 
52
 
53
+ print(encoder_name_or_path,decoder_name_or_path,)
54
+ model = VisionEncoderDecoderModel.from_encoder_decoder_pretrained(encoder_name_or_path,decoder_name_or_path)
55
 
56
 
57
  tokenizer = AutoTokenizer.from_pretrained(decoder._name_or_path)