a

streamlit_app.py

@@ -7,6 +7,8 @@ st.write(
 )
 
 model = ModelCatalog().load_model("bling-answer-tool")
-response = model.inference("My son is 21 years old.\nHow old is my son?")
-
-st.write("\nresponse: ", response)
+model = ModelCatalog().load_model("bling-answer-tool")
+prompt = "My son is 21 years old.\nHow old is my son?"
+response = model.inference(prompt)
+st.write("\nPrompt: ", prompt)
+st.write("\nResponse: ", response)