Update app.py

app.py

@@ -26,7 +26,7 @@ models_supported = {
         VisionEncoderDecoderModel.from_pretrained(
             "MohamedRashad/arabic-base-nougat",
             torch_dtype=torch.bfloat16,
-            attn_implementation={"decoder": "flash_attention_2", "encoder": "eager"},
+            # attn_implementation={"decoder": "flash_attention_2", "encoder": "eager"},
         ),
     ],
     "arabic-large-nougat": [
@@ -34,7 +34,7 @@ models_supported = {
         VisionEncoderDecoderModel.from_pretrained(
             "MohamedRashad/arabic-large-nougat",
             torch_dtype=torch.bfloat16,
-            attn_implementation={"decoder": "flash_attention_2", "encoder": "eager"},
+            # attn_implementation={"decoder": "flash_attention_2", "encoder": "eager"},
         ),
     ],
     # "mobser-small-v0.1": [