Update app.py

app.py

@@ -1,9 +1,15 @@
 import gradio as gr
 from transformers import pipeline
+import torch
+
+# Detect GPU availability and set the device
+device = 0 if torch.cuda.is_available() else -1
+print(f"Using device: {'GPU' if torch.cuda.is_available() else 'CPU'}")
 
 # Load the sentiment analysis pipeline from Hugging Face
 model = pipeline("sentiment-analysis", 
-                 model="nlptown/bert-base-multilingual-uncased-sentiment")
+                 model="nlptown/bert-base-multilingual-uncased-sentiment",
+                 device=device)
 
 # Function to get sentiment prediction
 def analyze_sentiment(text):