Hugging Face's logo

Spaces:
Shreyas94
/
SentinelAI102
Sleeping

App Files Community
Shreyas94 commited on
Commit
30f0e84
·
verified ·
1 Parent(s): 24df5af

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ logger = logging.getLogger(__name__)
13
 
14
  # Define device and load model and tokenizer
15
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
16
- MODEL_NAME = "microsoft/Phi-3-medium-128k-instruct"
17
 
18
  # Load model and tokenizer
19
  try:
 
13
 
14
  # Define device and load model and tokenizer
15
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
16
+ MODEL_NAME = "microsoft/Phi-3-mini-4k-instruct"
17
 
18
  # Load model and tokenizer
19
  try: