Update app.py

app.py

@@ -19,8 +19,9 @@ def greet(prompts, separator):
     from transformers.utils import move_cache
     from huggingface_hub import snapshot_download, login
 
-    LLM_MODEL_ID = "mistral-community/Mistral-7B-v0.2"
-    fp = snapshot_download(LLM_MODEL_ID)
+    LLM_MODEL_ID = "mozart-ai/llm-schema-extractor"
+    
+    fp = snapshot_download(LLM_MODEL_ID, token=os.environ['HF_TOKEN'])
     move_cache()
 
     model = LLM(fp)