Fix 'super_res_model is not defined' error

app.py

@@ -2,7 +2,6 @@ import gradio as gr
 from PIL import Image
 
 import _thread
-import time
 import torch
 import numpy as np
 from models.network_swinir import SwinIR as net
@@ -23,13 +22,14 @@ def load_model():
     super_res_model.load_state_dict(super_res_pretrained_model[param_key_g] if param_key_g in super_res_pretrained_model.keys() else super_res_pretrained_model, strict=True)
     super_res_model.eval()
 
+super_res_model=None
 _thread.start_new_thread(load_model, tuple())
 
 def predict(input_img):
     out = None
     
     # preprocess input
-    if(input_img is not None):
+    if(input_img is not None and super_res_model is not None):
         # model predict
         img_lq = input_img.astype(np.float32) / 255
         img_lq = np.transpose(img_lq if img_lq.shape[2] == 1 else img_lq[:, :, [2, 1, 0]], (2, 0, 1))  # HCW-BGR to CHW-RGB