app.py

import torch
import gradio as gr

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

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()