Upload app.py

app.py

@@ -20,17 +20,8 @@ import math
 
 DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-# --- compat shim for newer Phi-3/4 modeling files ---
-import transformers as _tf
-# Some HF model repos do: `from transformers.utils import LossKwargs`
-# Older/newer lib versions may not re-export it, so we provide a fallback.
-if not hasattr(_tf.utils, "LossKwargs"):
-    from typing import Dict, Any
-    class _LossKwargs(dict):
-        """Duck-typed placeholder; used only for type hints / Unpack at runtime."""
-        pass
-    _tf.utils.LossKwargs = _LossKwargs
-# -----------------------------------------------------
+
+
 
 
 # Set up logging