Upload app.py

app.py

@@ -301,7 +301,7 @@ elif tabs == 'Upload': #and count_system () != 1:
                     ]
         target_layer_dff = model.beit.layernorm
         target_layer_gradcam = model.beit.encoder.layer[-2].output
-        image_resized = image.resize((224, 224))
+        image_resized = image
         tensor_resized = transforms.ToTensor()(image_resized)
         outputs = model(**inputs)
         logits = outputs.logits
@@ -328,7 +328,7 @@ elif tabs == 'Upload': #and count_system () != 1:
             dic["name"] = x
             list_to_be_sorted.append(dic)
             data_base.append(y)
-        st.write(list_to_be_sorted)
+    
         if list_to_be_sorted[0]['name'] == "adenocarcinoma":
             dff_image.save(r"./Adenocarcinoma/{}".format(name_of_files_new[i]))
             image_path = name_of_files_new[i]
@@ -379,7 +379,7 @@ elif tabs == 'Upload': #and count_system () != 1:
     with col1:
         st.markdown("<h2 style='text-align: center; border: 2px solid #5370c6; border-radius: 5px; padding: 15px; background-color: white; color: black;' > Adenocarcinoma </h2>".format(centered_style), unsafe_allow_html=True)
         # Add the second subheader to the second column
-        folder_path = r"./Adenocarcinoma"
+        folder_path = r"./Adenocarcinoma/"
         image_files = [f for f in os.listdir(folder_path) if f.endswith('.png') or f.endswith('.jpg')]
         # Display the images in a loop
         for i in range(0, len(image_files), 2):
@@ -398,7 +398,7 @@ elif tabs == 'Upload': #and count_system () != 1:
                     count_classes.append("Adeno")
     with col2:
         st.markdown("<h2 style='text-align: center; border: 2px solid green; border-radius: 5px; padding: 15px; background-color: white; color: black;' > Normal </h2>".format(centered_style), unsafe_allow_html=True)
-        folder_path = r"./Normal"
+        folder_path = r"./Normal/"
         image_files = [f for f in os.listdir(folder_path) if f.endswith('.png') or f.endswith('.jpg')]
         # Display the images in a loop
         for i in range(0, len(image_files), 2):
@@ -422,7 +422,7 @@ elif tabs == 'Upload': #and count_system () != 1:
 
     with col5:
         st.markdown("<h2 style='text-align: center; border: 2px solid orange; border-radius: 5px; padding: 15px; background-color: white; color: black;' > Large cell carcinoma </h2>".format(centered_style), unsafe_allow_html=True)
-        folder_path = r"./Large cell carcinoma"
+        folder_path = r"./Large cell carcinoma/"
         image_files = [f for f in os.listdir(folder_path) if f.endswith('.png') or f.endswith('.jpg')]
         # Display the images in a loop
         for i in range(0, len(image_files), 2):
@@ -441,7 +441,7 @@ elif tabs == 'Upload': #and count_system () != 1:
                     count_classes.append("Large")
     with col6:
         st.markdown("<h2 style='text-align: center; border: 2px solid #f16565; border-radius: 5px; padding: 15px; background-color: white; color: black;' >  Squamous cell carcinoma </h2>".format(centered_style), unsafe_allow_html=True)
-        folder_path = r"./Squamous cell carcinoma"
+        folder_path = r"./Squamous cell carcinoma/"
         image_files = [f for f in os.listdir(folder_path) if f.endswith('.png') or f.endswith('.jpg')]
         # Display the images in a loop
         for i in range(0, len(image_files), 2):