Update app.py

app.py

@@ -11,7 +11,6 @@ For more information on `huggingface_hub` Inference API support, please check th
 """
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
-# Load model and tokenizer once
 model = load_model()
 tokenizer = load_tokenizer()
 
@@ -40,7 +39,10 @@ def gradio_shap(context):
         plt.tight_layout()
         return fig
     except Exception as e:
-        return f"SHAP explanation failed: {e}"
+        fig, ax = plt.subplots(figsize=(8, 2))
+        ax.text(0.5, 0.5, f"SHAP explanation failed:\n{e}", ha='center', va='center', wrap=True, fontsize=12)
+        ax.axis('off')
+        return fig
 
 with gr.Blocks() as demo:
     gr.Markdown("# 🧠 C3QG – Context-Controlled Question Generation with FLAN-T5")
@@ -62,6 +64,5 @@ with gr.Blocks() as demo:
         inputs=context,
         outputs=shap_output
     )
-
 if __name__ == "__main__":
     demo.launch()