Update app.py

app.py

@@ -2,7 +2,7 @@
 import gradio as gr
 import os
 import torch
-
+from model import TinyCNN
 from timeit import default_timer as timer
 from typing import Tuple, Dict