hotmemeh commited on
Commit
741e6dd
·
verified ·
1 Parent(s): cd7e456

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import torch
4
 
5
  # Auto-select model based on device
6
  if torch.cuda.is_available():
7
- MODEL_NAME = "tiiuae/falcon-7b-instruct" # GPU model
8
  device = 0
9
  else:
10
  MODEL_NAME = "gpt2" # CPU fallback
 
4
 
5
  # Auto-select model based on device
6
  if torch.cuda.is_available():
7
+ MODEL_NAME = "darkc0de/XortronCriminalComputingConfig"
8
  device = 0
9
  else:
10
  MODEL_NAME = "gpt2" # CPU fallback