Update app.py

app.py

@@ -2,12 +2,14 @@ import gradio as gr
 from model import config
 from PIL import Image
 # import pickle
+import json
 # import numpy as np
 # from fastapi import FastAPI,Response
 # from sklearn.metrics import accuracy_score, f1_score
 # import prometheus_client as prom
 # import pandas as pd
 # import uvicorn
+import os
 from transformers import VisionEncoderDecoderModel,pipeline, ViTImageProcessor, AutoTokenizer
 import torch
 
@@ -40,14 +42,17 @@ import torch
 
 #     f1_metric.set(f1)
 
+from model.config import encoder,decoder
+
+# with open('model/config.json', 'r') as file:
+#     config = json.load(file)
 
-    
 
 print(encoder._name_or_path,decoder._name_or_path,)
-model = VisionEncoderDecoderModel.from_encoder_decoder_pretrained(config.encoder._name_or_path, config.decoder._name_or_path)
+model = VisionEncoderDecoderModel.from_encoder_decoder_pretrained(encoder._name_or_path,decoder._name_or_path)
 
 
-tokenizer = AutoTokenizer.from_pretrained(config.decoder._name_or_path)
+tokenizer = AutoTokenizer.from_pretrained(decoder._name_or_path)
 tokenizer.pad_token = tokenizer.unk_token