Update app.py

app.py

@@ -13,6 +13,7 @@ import gradio as gr
 from PIL import Image
 import numpy as np
 from matplotlib import pyplot as plt
+from model import dehazeformer_t
 
 # Get cpu or gpu device for training.
 device = "cuda" if torch.cuda.is_available() else "mps" if torch.backends.mps.is_available() else "cpu"