Update app.py

app.py

@@ -109,52 +109,39 @@ topic_model_2_output = topic_model_2.get_topic_info(topic[0])["Representation"][
 ]
 ###############################################################################
 
-print()
-
-print("-" * 150)
-print("vqa_pipeline_output = ", vqa_pipeline_output)
-print("bbu_pipeline_output =", bbu_pipeline_output)
-print("facebook_model_output =", facebook_model_output)
-print("gpt2_pipeline_output =", gpt2_pipeline_output)
-print("topic_model_1_output =", topic_model_1_output)
-print("topic_model_2_output =", topic_model_2_output)
-
-print()
-
-print("-" * 150)
-print("SUMMARY")
-print("-" * 7)
-print("Your Image:", image)
-print("Your Question:", question)
-print("-" * 100)
-print(
-    "1. Highest Predicted Answer For Your Question:",
-    vqa_pipeline_output["answer"],
-    "\n",
-    )
-print(text)
-print(
-    "2. Highest Predicted Sequence On [MASK] Based on 1.:",
-    bbu_pipeline_output[0]["sequence"],
-    "\n",
-)
-print(
-    "3. Conversation Based On Previous Answer Based on 2.:",
-    facebook_model_output,
-    "\n",
-)
-print(
-    "4. Text Generated Based On Previous Answer Based on 3.:",
-    gpt2_pipeline_output,
-    "\n",
-)
-print(
-    "5. Highest Predicted Topic Model_1 For Previous The Answer Based on 4.:",
-    topic_model_1_output,
-    "\n",
-)
-print(
-    "6. Highest Predicted Topic Model_2 For Previous The Answer Based on 4.:",
-    topic_model_2_output,
-)
-print("-" * 150)
+st.write("-" * 150)
+st.write("vqa_pipeline_output =", vqa_pipeline_output)
+st.write("bbu_pipeline_output =", bbu_pipeline_output)
+st.write("facebook_model_output =", facebook_model_output)
+st.write("gpt2_pipeline_output =", gpt2_pipeline_output)
+st.write("topic_model_1_output =", topic_model_1_output)
+st.write("topic_model_2_output =", topic_model_2_output)
+st.write("-" * 150)
+
+st.write("SUMMARY")
+st.subheader("Your Image:")
+st.image(image, caption="Your Image", use_column_width=True)
+st.subheader("Your Question:")
+st.write(question)
+st.write("-" * 100)
+
+st.subheader("1. Highest Predicted Answer For Your Question:")
+st.write(vqa_pipeline_output["answer"])
+st.write(text)
+
+st.subheader("2. Highest Predicted Sequence On [MASK] Based on 1.:")
+st.write(bbu_pipeline_output[0]["sequence"])
+
+st.subheader("3. Conversation Based On Previous Answer Based on 2.:")
+st.write(facebook_model_output)
+
+st.subheader("4. Text Generated Based On Previous Answer Based on 3.:")
+st.write(gpt2_pipeline_output)
+
+st.subheader("5. Highest Predicted Topic Model_1 For Previous The Answer Based on 4.:")
+st.write(topic_model_1_output)
+
+st.subheader("6. Highest Predicted Topic Model_2 For Previous The Answer Based on 4.:")
+st.write(topic_model_2_output)
+
+st.write("-" * 150)