Update app.py

app.py

@@ -92,7 +92,8 @@ def adr_predict(x):
     return {"Severe Reaction": float(scores.numpy()[1]), "Non-severe Reaction": float(scores.numpy()[0])}, shap_scores
     # , word_attributions ,scores
 
-def main(text):
+def main(prob1):
+    text = str(prob1).lower()
     obj = adr_predict(text)
     return obj[0],obj[1]
     # ,obj[2]
@@ -106,7 +107,7 @@ with gr.Blocks(title=title) as demo:
     gr.Markdown(f"## {title}")
     gr.Markdown(description1)
     gr.Markdown("""---""")
-    text = gr.Textbox(label="Enter Your Text Here:",lines=2, placeholder="Type it here ...")
+    prob1 = gr.Textbox(label="Enter Your Text Here:",lines=2, placeholder="Type it here ...")
     submit_btn = gr.Button("Analyze")
 
 
@@ -126,7 +127,7 @@ with gr.Blocks(title=title) as demo:
 
     submit_btn.click(
         main,
-        [text],
+        [prob1],
         [label
          # ,intp
          ,interpretation
@@ -134,7 +135,7 @@ with gr.Blocks(title=title) as demo:
     )
 
     gr.Markdown("### Click on any of the examples below to see to what extent they contain resilience messaging:")
-    gr.Examples([["I have minor pain."],["I have severe pain."]], [text], [label
+    gr.Examples([["I have minor pain."],["I have severe pain."]], [prob1], [label
                                                                            # ,intp
                                                                            ,interpretation
                                                                           ], main, cache_examples=True)