fix

app.py

@@ -7,7 +7,7 @@ import numpy as np
 from SUPIR.util import create_SUPIR_model, load_QF_ckpt
 from PIL import Image
 from llava.llava_agent import LLavaAgent
-from model_files.CKPT_PTH import LLAVA_MODEL_PATH
+from CKPT_PTH import LLAVA_MODEL_PATH
 import einops
 import copy
 import time
@@ -24,11 +24,11 @@ else:
     raise ValueError('Currently support CUDA only.')
 
 # Load SUPIR model
-model, default_setting = create_SUPIR_model('model_files/options/SUPIR_v0.yaml', SUPIR_sign='Q', load_default_setting=True)
+model, default_setting = create_SUPIR_model('options/SUPIR_v0.yaml', SUPIR_sign='Q', load_default_setting=True)
 model = model.to(SUPIR_device)
 model.first_stage_model.denoise_encoder_s1 = copy.deepcopy(model.first_stage_model.denoise_encoder)
 model.current_model = 'v0-Q'
-ckpt_Q, ckpt_F = load_QF_ckpt('model_files/options/SUPIR_v0.yaml')
+ckpt_Q, ckpt_F = load_QF_ckpt('options/SUPIR_v0.yaml')
 
 # Load LLaVA
 llava_agent = LLavaAgent(LLAVA_MODEL_PATH, device=LLaVA_device, load_8bit=True, load_4bit=False)