Update app.py

app.py

@@ -55,7 +55,7 @@ if __name__ == "__main__":
             img = load_image(file, img_size)
             pred = model.predict(img)
             output = classes[round(pred[0][0])]
-            st.subheader(f"The image is {output}")
+            st.subheader(f"The image is {output} for TB")
         else:
             st.write("Please upload an image first")