Update modeling_lang.py

modeling_lang.py

@@ -1,3 +1,15 @@
+import subprocess
+import sys
+
+# Ensure floret is installed
+try:
+    import floret
+except ImportError:
+    print("Installing floret...")
+    subprocess.check_call([sys.executable, "-m", "pip", "install", "floret"])
+    subprocess.check_call([sys.executable, "-m", "pip", "install", "huggingface_hub"])
+    import floret
+
 import torch
 import torch.nn as nn
 from transformers import PreTrainedModel