Paula Ayman commited on
Commit
df632f9
·
1 Parent(s): ea09339
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
3
  import torch
4
 
5
  # Pick any model you want to test (small for free Spaces)
6
- MODEL_ID = "microsoft/phi-2"
7
 
8
  # Load tokenizer + model
9
  tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
 
3
  import torch
4
 
5
  # Pick any model you want to test (small for free Spaces)
6
+ MODEL_ID = "LiquidAI/LFM2-1.2B"
7
 
8
  # Load tokenizer + model
9
  tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)