app.py

@@ -4,6 +4,7 @@ import os
 import time
 from pathlib import Path
 
+import gradio as gr
 import torch
 from loguru import logger
 from transformers import AutoModel, AutoTokenizer