barbaroo commited on
Commit
343c3c7
1 Parent(s): 6a7f812

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -0
app.py CHANGED
@@ -1,3 +1,5 @@
 
 
1
  import gradio as gr
2
  from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
3
  import nltk
@@ -76,3 +78,10 @@ iface = gr.Interface(
76
  # Launch the interface
77
  iface.launch()
78
 
 
 
 
 
 
 
 
 
1
+
2
+ '''
3
  import gradio as gr
4
  from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
5
  import nltk
 
78
  # Launch the interface
79
  iface.launch()
80
 
81
+ '''
82
+
83
+ import torch
84
+ print("CUDA available:", torch.cuda.is_available())
85
+ print("CUDA device count:", torch.cuda.device_count())
86
+ print("CUDA current device:", torch.cuda.current_device())
87
+ print("CUDA device name:", torch.cuda.get_device_name(0))