helloperson123 commited on
Commit
7cfa400
·
verified ·
1 Parent(s): 9e876b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ CORS(app)
10
  # ----------------------------
11
  # Model (UNGATED, FREE)
12
  # ----------------------------
13
- MODEL_NAME = "microsoft/phi-3-mini-4k-instruct"
14
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
15
 
16
  device = torch.device("cpu")
 
10
  # ----------------------------
11
  # Model (UNGATED, FREE)
12
  # ----------------------------
13
+ model = "microsoft/phi-3-mini-4k-instruct"
14
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
15
 
16
  device = torch.device("cpu")