Update app.py

app.py

@@ -97,9 +97,9 @@ if torch.cuda.is_available():
     if "A100" not in gpu_name and "L4" not in gpu_name:
         raise RuntimeError(f"This notebook requires an A100 or L4 GPU. Found: {gpu_name}")
     elif "L4" in gpu_name:
-        print("Warning: L4 is supported, but A100 is recommended for faster inference.")
+        print("Warning: L4 or L40S is supported, but A100 is recommended for faster inference.")
 else:
-    raise RuntimeError("No CUDA-compatible GPU found. An A100 or L4 GPU is required.")
+    raise RuntimeError("No CUDA-compatible GPU found. An A100, L4 or L40S GPU is required.")
 
 
 GPU_TO_VRAM_PARAMS = {