Update app.py

app.py

@@ -112,7 +112,7 @@ def predictor(texts):
 
 def lime_explain(code_txt):
   class_names = ['Without OpenMP', 'With OpenMP']
-  SAMPLES = 40
+  SAMPLES = 35
   exp = []
 
   if predict(code_txt)[0] == without_omp_str:
@@ -124,6 +124,13 @@ def lime_explain(code_txt):
   return gr.update(visible=True, value=exp.as_pyplot_figure())
 
 
+def lime_title(code_txt):
+  if predict(code_txt)[0] == without_omp_str:
+      return gr.update(visible=False)
+
+  return gr.update(visible=True)
+    
+
 def activate_c(lang_pred):
   langs = lang_pred.split('\n')
   langs = {lang[5:lang.find(':')]:float(lang[lang.find(':')+1:]) for lang in langs}
@@ -198,6 +205,7 @@ with gr.Blocks() as pragformer_gui:
             private = gr.Textbox(label="Data-sharing attribute clause- private", visible=False)
             reduction = gr.Textbox(label="Data-sharing attribute clause- reduction", visible=False)
 
+        explain_title = gr.Markdown("## LIME Explainability", visible=False)
         explanation = gr.Plot(visible=False)
 
 
@@ -209,6 +217,7 @@ with gr.Blocks() as pragformer_gui:
     submit_btn.click(fn=is_private, inputs=code_in, outputs=private)
     submit_btn.click(fn=is_reduction, inputs=code_in, outputs=reduction)
     submit_btn.click(fn=lime_explain, inputs=code_in, outputs=explanation)
+    submit_btn.click(fn=lime_title, inputs=code_in, outputs=explain_title)
     sample_btn.click(fn=fill_code, inputs=drop, outputs=[pragma, code_in])
 
     gr.Markdown(