Update app.py

app.py

@@ -3,6 +3,7 @@ import gradio as gr
 import os
 import torch
 
+from torch import nn
 from model import create_resnet50_model
 from timeit import default_timer as timer
 from typing import Tuple, Dict