Requirements and mdm wheel

README.md

@@ -1,6 +1,6 @@
 ---
-title: Mdm
-emoji: 📚
+title: Matryoshka
+emoji: 🪆
 colorFrom: blue
 colorTo: blue
 sdk: gradio

app.py

@@ -24,6 +24,8 @@ from ml_mdm import helpers, reader
 from ml_mdm.config import get_arguments, get_model, get_pipeline
 from ml_mdm.language_models import factory
 
+os.system("pip install --upgrade ./ml_mdm-1.0-py3-none-any.whl")
+
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 # Download destination

requirements.txt

@@ -0,0 +1 @@
+transformers