random2222 commited on
Commit
5f7ceab
·
verified ·
1 Parent(s): 9a29aa2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ def load_model():
17
  print(f"Using device: {device}")
18
 
19
  # Load the Phi-2 model
20
- model_id = "microsoft/phi-2"
21
 
22
  print("Loading Phi-2 model and tokenizer...")
23
  tokenizer = AutoTokenizer.from_pretrained(model_id)
 
17
  print(f"Using device: {device}")
18
 
19
  # Load the Phi-2 model
20
+ model_id = "microsoft/phi-1_5"
21
 
22
  print("Loading Phi-2 model and tokenizer...")
23
  tokenizer = AutoTokenizer.from_pretrained(model_id)