Update app.py

app.py

@@ -9,7 +9,7 @@ import streamlit as st
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 # Load the trained model from the Hugging Face Hub
-model = AutoModel.from_pretrained('dhhd255/parkinsons_20epochs')
+model = AutoModel.from_pretrained('dhhd255/EfficientNet_ParkinsonsPred')
 
 # Move the model to the device
 model = model.to(device)
@@ -31,19 +31,7 @@ st.markdown("""
  .parkinsons {
  color: #C30000;
  }
- .caption_c{
-  position: relative;
-  display: flex;
-  flex-directon: column;
-  align-items: center;
-  top: calc(99vh - 370px);
- }
- .caption {
-  
- text-align: center;
- color: #646464;
- font-size: 14px;
- }
+
  </style>
 """, unsafe_allow_html=True)
 
@@ -83,6 +71,3 @@ if uploaded_file is not None:
         else:
             col2.markdown('<span class="result healthy">Predicted class: Healthy</span>', unsafe_allow_html=True)
             col2.caption(f'{confidence*100:.0f}% sure')
-
-# Add a caption at the bottom of the page
-st.markdown('<div class="caption_c"><p class="caption">Made with love by Jayant</p></div>', unsafe_allow_html=True)