Update app.py

app.py

@@ -5,14 +5,15 @@
 # ------------------------------
 # 0) Imports & environment
 # ------------------------------
-import os, 
+import os 
 os.environ.setdefault("HF_HUB_ENABLE_HF_TRANSFER", "1")
-import io, csv, time, json, base64, re, zipfile
-from typing import List, Tuple, Dict, Any
-
-# Persist model caches between restarts
 os.environ.setdefault("HF_HOME", "/home/user/.cache/huggingface")
 os.makedirs(os.environ["HF_HOME"], exist_ok=True)
+import io, csv, time, json, base64, re, zipfile
+from typing import Generator, List, Tuple, Dict, Any
+from threading import Thread
+# Persist model caches between restarts
+
 
 import gradio as gr
 from PIL import Image