gpuggraph3d / app.py
patrjfjfkfr's picture
Update app.py
9c89958 verified
raw
history blame contribute delete
281 Bytes
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()