Update app.py

app.py

@@ -22,10 +22,18 @@ import torch
 import matplotlib.pyplot as plt
 from torchvision.transforms import ToTensor, ToPILImage
 
+if torch.cuda.is_available():
+    device = 0  
+    print("CUDA is available. Using GPU.")
+else:
+    device = -1  
+    print("CUDA is not available. Using CPU.")
+
+
     
 client = OpenAI()
 
-pipe = pipeline("zero-shot-image-classification", model="patrickjohncyh/fashion-clip")
+pipe = pipeline("zero-shot-image-classification", model="patrickjohncyh/fashion-clip", device=device)
 
 color_file_path = 'color_config.json'
 attributes_file_path = 'attributes_config.json'