BMC bug fix

app.py

@@ -76,7 +76,20 @@ def main():
     st.title("Image Classification")
     st.write("This is a simple web app to test and compare different image classifier models using Hugging Face's image-classification pipeline.")
     st.write("From time to time more models will be added to the list. If you want to add a model, please open an issue on the GitHub repository.")
-    st.markdown("<img src='https://helloimjessa.files.wordpress.com/2021/06/bmc-button.png?w=1024' width='15%'>", unsafe_allow_html=True)
+    
+    # Buy me a Coffee Setup
+    bmc_link = "https://www.buymeacoffee.com/nuno.tome" 
+    image_url = "https://helloimjessa.files.wordpress.com/2021/06/bmc-button.png?w=150" # Image URL
+    image_size = "150px" # Image size
+    #image_link_markdown = f"<img src='{image_url}' width='25%'>"
+    image_link_markdown = f"[![Buy Me a Coffee]({image_url})]({bmc_link})"
+
+    #image_link_markdown = f"[![Buy Me a Coffee]({image_url})]({bmc_link})" # Create a clickable image link
+
+    st.markdown(image_link_markdown, unsafe_allow_html=True) # Display the image link
+    # Buy me a Coffee Setup
+    
+    #st.markdown("<img src='https://helloimjessa.files.wordpress.com/2021/06/bmc-button.png?w=1024' width='15%'>", unsafe_allow_html=True)
   
     input_image = st.file_uploader("Upload Image")
     shosen_model = st.selectbox("Select the model to use",  MODELS)
@@ -84,11 +97,11 @@ def main():
     
     if input_image is not None:
         image_to_classify = Image.open(input_image)
-        st.image(image_to_classify, caption="Uploaded Image", use_column_width=True)
+        st.image(image_to_classify, caption="Uploaded Image")
         if st.button("Classify"):
             image_to_classify = Image.open(input_image)
             classification_obj1 =[]
-            avable_models = st.selectbox
+            #avable_models = st.selectbox
             
             classification_result = classify(image_to_classify, shosen_model)
             classification_obj1.append(classification_result)