Update app.py

app.py

@@ -8,7 +8,11 @@ import random
 # Load the InstructPix2Pix model
 model_id = "timbrooks/instruct-pix2pix"
 pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-pipe = pipe.to("cpu")
+
+# Check if a GPU is available, otherwise fallback to CPU
+device = "cuda" if torch.cuda.is_available() else "cpu"
+pipe = pipe.to(device)  # Move the model to the appropriate device
+
 
 # Load the translation model (from Arabic to English)
 translation_model_name = 'Helsinki-NLP/opus-mt-ar-en'