test / app.py
rotemvahava's picture
Rename test to app.py
4817daa verified
Raw
History Blame Contribute Delete
233 Bytes
import spaces
import torch
import gradio as gr
@spaces.GPU
def check():
return f"CUDA available: {torch.cuda.is_available()}\nDevice: {torch.cuda.get_device_name(0)}"
gr.Interface(fn=check, inputs=[], outputs="text").launch()