Akjava commited on
Commit
511b690
Β·
verified Β·
1 Parent(s): 958f07a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -39,7 +39,7 @@ def init():
39
  print("The model is on a CPU")
40
 
41
  print(f"text_generator.device='{text_generator.device}")
42
- if text_generator.device.strip() == 'cuda':
43
  print("The pipeline is using a GPU")
44
  else:
45
  print("The pipeline is using a CPU")
 
39
  print("The model is on a CPU")
40
 
41
  print(f"text_generator.device='{text_generator.device}")
42
+ if str(text_generator.device).strip() == 'cuda':
43
  print("The pipeline is using a GPU")
44
  else:
45
  print("The pipeline is using a CPU")