Update app.py
Browse files
app.py
CHANGED
|
@@ -4,7 +4,7 @@ import torch
|
|
| 4 |
|
| 5 |
# Auto-select model based on device
|
| 6 |
if torch.cuda.is_available():
|
| 7 |
-
|
| 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
|