Update app.py

app.py

@@ -5,6 +5,7 @@ import logging
 import plotly.express as px
 from sklearn.ensemble import IsolationForest  # For anomaly detection
 from transformers import pipeline
+import torch  # For GPU availability check
 
 # Configure logging for debugging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -12,8 +13,10 @@ logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(
 # Preload Hugging Face summarization model at startup
 logging.info("Preloading Hugging Face model...")
 try:
-    summarizer = pipeline("text2text-generation", model="google/flan-t5-base", device=0 if gr.cuda.is_available() else -1)  # Use GPU if available
-    logging.info("Hugging Face model preloaded successfully")
+    # Check for GPU availability using torch.cuda
+    device = 0 if torch.cuda.is_available() else -1
+    summarizer = pipeline("text2text-generation", model="google/flan-t5-base", device=device)  # Use GPU if available
+    logging.info(f"Hugging Face model preloaded successfully on device: {'GPU' if device == 0 else 'CPU'}")
 except Exception as e:
     logging.error(f"Failed to preload model: {str(e)}")
     raise e
@@ -166,7 +169,7 @@ def process_logs(file_obj, progress=gr.Progress()):
             logging.error("Unsupported file format")
             return "Unsupported file format. Please upload a CSV file.", None, None, None, None, None
 
-        # Use pandas to load CSV (removed dask dependency)
+        # Use pandas to load CSV
         progress(0.05, "Loading CSV file...")
         try:
             df = pd.read_csv(file_name)