Upload app.py

app.py

@@ -21,6 +21,19 @@ import time
 
 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
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)