sandz7 commited on
Commit
1dd4ded
Β·
1 Parent(s): 0a35e8d

fixed (0) in check_cuda()

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -209,7 +209,7 @@ def loki_generation(input_text: str,
209
 
210
  def check_cuda():
211
  if torch.cuda.is_available():
212
- return f"GPU Being Used: {torch.cuda.get_device_name[0]}"
213
  else:
214
  return "No GPU is being used right now."
215
 
 
209
 
210
  def check_cuda():
211
  if torch.cuda.is_available():
212
+ return f"GPU Being Used: {torch.cuda.get_device_name(0)}"
213
  else:
214
  return "No GPU is being used right now."
215