rainbow-canvas / app.py
homer-meng's picture
Update app.py
7cea500
raw
history blame contribute delete
378 Bytes
import streamlit as st
import torch
import requests
st.title("Hugging Face Space GPU Info")
# Check if GPU is available
gpu_available = torch.cuda.is_available()
if gpu_available:
gpu_name = torch.cuda.get_device_name(0)
st.write(f"GPU available: {gpu_name}")
else:
st.write("No GPU available")
r = requests.get("https://www.google.com")
st.write(r.status_code)