Update app.py

app.py

@@ -2,6 +2,9 @@ import gradio as gr
 import torch
 
 model_map = torch.hub.load('nateraw/image-generation:main', 'model_map')
+on_gpu = torch.cuda.is_available()
+
+print(f"GPU enabled? - {'🔴' if not on_gpu else '🟢'}")
 
 class InferenceWrapper:
     def __init__(self, model):