Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from model.config import *
+from model import config
 from PIL import Image
 # import pickle
 # import numpy as np
@@ -41,12 +41,13 @@ import torch
 #     f1_metric.set(f1)
 
 
+    
 
 print(encoder._name_or_path,decoder._name_or_path,)
-model = VisionEncoderDecoderModel.from_encoder_decoder_pretrained(encoder._name_or_path, decoder._name_or_path)
+model = VisionEncoderDecoderModel.from_encoder_decoder_pretrained(config.encoder._name_or_path, config.decoder._name_or_path)
 
 
-tokenizer = AutoTokenizer.from_pretrained(decoder._name_or_path)
+tokenizer = AutoTokenizer.from_pretrained(config.decoder._name_or_path)
 tokenizer.pad_token = tokenizer.unk_token