Update app.py

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 from PIL import Image
 from transformers import pipeline
 import io
+import torch  # Import PyTorch
 
 # --- Configuration ---
 # Specify the model
@@ -14,7 +15,7 @@ def load_pipeline():
     """Loads the VQA pipeline."""
     try:
         # Explicitly set device if CUDA is available, otherwise use CPU
-        device = 0 if any(tf.config.list_physical_devices('GPU')) else -1
+        device = 0 if torch.cuda.is_available() else -1  # Use torch.cuda
         vqa_pipeline = pipeline("visual-question-answering", model=MODEL_NAME, device=device) # Add device
         return vqa_pipeline
     except Exception as e: