Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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))
|