Handling errors.

app.py

@@ -27,8 +27,16 @@ try:
         "D":d,
         "E":e
     }
-    answer = pipeline.give_the_best_answer(mi_dict)
-    st.write(mi_dict)
-    st.write(answer)
+
+    try:    
+        answer = pipeline.give_the_best_answer(mi_dict)
+        st.write(mi_dict)
+        st.write(answer)
+    except Exception as e:
+        st.error("something bad happend to the model")
+        st.error(e)
+
 except:
-    st.error("Your question doesn't have the required format. Please, correct it.")
+    st.error("Your question doesn't have the required format. Please, correct it.")
+    st.error(e)
+

mi_clase.py

@@ -129,8 +129,8 @@ class Pipeline:
     #---- retriever + generator
 
     def give_the_best_answer(self,dict_with_all_the_info):
-        a = self.get_context_from_text(my_dict["prompt"])
-        b = self.create_tokens(my_dict,a)
+        a = self.get_context_from_text(dict_with_all_the_info["prompt"])
+        b = self.create_tokens(dict_with_all_the_info,a)
         c = self.infer_answer(b)
         d = self.convert_to_letter(int(c[0][0]))
         #print("\nThe answer is ",)