manoskary commited on
Commit
19e0f81
·
verified ·
1 Parent(s): 632832e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ logger = logging.getLogger(__name__)
19
 
20
  # Initialize the NotaGen tool
21
  # This will use GPU if available in the HuggingFace Space
22
- device = "cuda" if torch.cuda.is_available() else "cpu"
23
  logger.info(f"Initializing NotaGen tool on device: {device}")
24
 
25
  try:
 
19
 
20
  # Initialize the NotaGen tool
21
  # This will use GPU if available in the HuggingFace Space
22
+ device = "cuda"
23
  logger.info(f"Initializing NotaGen tool on device: {device}")
24
 
25
  try: