Update app.py

app.py

@@ -40,7 +40,8 @@ if index_id not in pinecone.list_indexes():
 
 index = pinecone.Index(index_id)
 
-device = 'cpu'
+device = 'cuda' if torch.cuda.is_available() else 'cpu'
+print(f"Using '{device}' device...")
 
 # init all of the models and move them to a given GPU
 pipe = StableDiffusionPipeline.from_pretrained(