Update app.py

app.py

@@ -28,6 +28,9 @@ from transformers import CLIPTextModelWithProjection, CLIPTextModel
 from transformers import Blip2Processor, Blip2ForConditionalGeneration
 from transformers import Phi3ForCausalLM
 from transformers import pipeline
+import torchvision
+
+torchvision.set_image_backend('accimage')
 
 from transformers import InstructBlipProcessor, InstructBlipForConditionalGeneration
 torch.backends.cuda.matmul.allow_tf32 = False
@@ -38,8 +41,8 @@ torch.backends.cudnn.deterministic = False
 torch.backends.cudnn.benchmark = False
 #torch.backends.cuda.preferred_blas_library="cublas"
 # torch.backends.cuda.preferred_linalg_library="cusolver"
-
 torch.set_float32_matmul_precision("highest")
+
 os.putenv("HF_HUB_ENABLE_HF_TRANSFER","1")
 os.environ["SAFETENSORS_FAST_GPU"] = "1"