commit

app.py

@@ -23,6 +23,7 @@ import spaces
 import numpy as np
 import numpy.typing as npt
 import torch
+from torch import nn
 import gradio as gr
 from lxml.html import fromstring
 from huggingface_hub import hf_hub_download