Update app.py

app.py

@@ -12,7 +12,7 @@ def main():
         if is_valid_filetype(uploaded_file):
             st.success("File uploaded successfully!")
             # Process the file here
-            st.write(check_file_type(uploaded_file))
+            st.write(get_features(uploaded_file))
         else:
             st.error("Invalid file type. Please upload an EML or TXT file.")
 
@@ -20,7 +20,6 @@ def main():
 def upload_file():
         filepath = 'email files/' + file.filename
         return render_template('home.html',
-                               content=check_file_type(file),
                                features = get_features(filepath),
                                pre_content=predict_content(text_feature(filepath)),
                                pre_tag=predict_html(html_tags_feature(filepath)),