gpu avail

app.py

@@ -1,13 +1,17 @@
 import gradio as gr
 from transformers import pipeline
+import torch
 
-# Load an available model for text generation
-model = pipeline("text-generation", model="gpt2")
+# Check if a GPU is available
+device = 0 if torch.cuda.is_available() else -1
+print("Using GPU" if device == 0 else "Using CPU")
+
+# Load the model on the GPU if available
+model = pipeline("text-generation", model="gpt2", device=device)
 
 def generate_text(prompt):
     return model(prompt, max_length=50)[0]["generated_text"]
 
-# Custom interface with a textbox and description
 interface = gr.Interface(
     fn=generate_text,
     inputs=gr.Textbox(label="Enter your prompt here"),