File size: 339 Bytes
921d811
 
 
 
7463dc4
 
921d811
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
import torch
import gradio as gr

print("CUDA Version:", torch.version.cuda)
print(torch.version.cuda)
print(torch.cuda.is_available())

zero = torch.Tensor([0]).cuda()
print(zero.device)

def my_function(input_text):
    return f"Received: {input_text}"

iface = gr.Interface(fn=my_function, inputs="text", outputs="text")
iface.launch()