Update app.py

app.py

@@ -1,13 +1,13 @@
-import torch
 import streamlit as st
-from diffusers import StableDiffusionPipeline
-from PIL import Image
+import torch
 
-pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
-pipe = pipe.to("cuda")
+st.title("Hugging Face Space GPU Info")
 
-prompt = st.text_area("Enter a prompt: ")
-image = pipe(prompt).images[0]
+# Check if GPU is available
+gpu_available = torch.cuda.is_available()
 
-image = Image.fromarray(image)
-image.show()
+if gpu_available:
+    gpu_name = torch.cuda.get_device_name(0)
+    st.write(f"GPU available: {gpu_name}")
+else:
+    st.write("No GPU available")