9c89958
c30bba4
d02cdd2
c30bba4
|
|
import torch
import gradio as gr
def check_gpu(_):
if torch.cuda.is_available():
return f"β
GPU detected: {torch.cuda.get_device_name(0)}"
else:
return "β No GPU yet. Try refreshing."
gr.Interface(fn=check_gpu, inputs="text", outputs="text").launch() |