Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -46,12 +46,12 @@ import torch
|
|
46 |
|
47 |
with open("model/config.json") as f:
|
48 |
n=json.load(f)
|
49 |
-
|
50 |
-
|
51 |
|
52 |
|
53 |
-
print(
|
54 |
-
model = VisionEncoderDecoderModel.from_encoder_decoder_pretrained(
|
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)
|