Restore CXR Transformers 4 compatibility

README.md

@@ -4,6 +4,7 @@ emoji: 🫁
 colorFrom: blue
 colorTo: indigo
 sdk: gradio
+sdk_version: 5.49.1
 app_file: app.py
 license: mit
 pinned: true
@@ -200,4 +201,4 @@ Ideal for research in:
 * Powered by **Gradio** and **Hugging Face Transformers**
 * Based on and expanded from the **Token-Attention-Viewer** project
 🔗 [https://github.com/devMuniz02/Image-Attention-Visualizer](https://github.com/devMuniz02/Image-Attention-Visualizer)
-* Created as part of a thesis on **efficient and explainable multimodal medical AI**
+* Created as part of a thesis on **efficient and explainable multimodal medical AI**

requirements.txt

@@ -1,4 +1,5 @@
-transformers==5.8.0
+transformers==4.57.1
+gradio==5.49.1
 torch==2.9.0
 torchvision==0.24.0
 fsspec==2025.10.0

utils/modifiedGPT2.py

@@ -359,7 +359,7 @@ class GPT2ModelModified(GPT2Model):
 
         causal_mask = create_causal_mask(
             config=self.config_causal,
-            inputs_embeds=inputs_embeds,
+            input_embeds=inputs_embeds,
             attention_mask=attention_mask,
             cache_position=cache_position,
             past_key_values=past_key_values,
@@ -709,4 +709,4 @@ def create_decoder(attention = "sdpa"):
     decoder = GPT2LMHeadModelModified.from_pretrained("gpt2", config=config)
     decoder.config._attn_implementation = attention
     decoder = expand_gpt2_positional_embeddings(decoder, new_max_positions=new_max_positions, mode="linear")
-    return decoder
+    return decoder