Update app.py

app.py

@@ -14,6 +14,22 @@ model = AutoModel.from_pretrained('dhhd255/parkinsons_pred0.1')
 # Move the model to the device
 model = model.to(device)
 
+# Add custom CSS to use the Inter font and define custom classes for healthy and parkinsons results
+st.markdown("""
+    <style>
+        @import url('https://fonts.googleapis.com/css2?family=Inter&display=swap');
+        body {
+            font-family: 'Inter', sans-serif;
+        }
+        .healthy {
+            color: #007E3F;
+        }
+        .parkinsons {
+            color: #C30000;
+        }
+    </style>
+""", unsafe_allow_html=True)
+
 st.title("Parkinson's Disease Prediction")
 
 uploaded_file = st.file_uploader("Choose an image...", type=["png", "jpg", "jpeg"])
@@ -44,6 +60,6 @@ if uploaded_file is not None:
         logits = feature_reducer(logits)
         predicted_class = torch.argmax(logits, dim=1).item()
         if(predicted_class == 0):
-            col2.subheader('Predicted class: Parkinson\'s')
+            col2.markdown('<span class="parkinsons">Predicted class: Parkinson\'s</span>', unsafe_allow_html=True)
         else:
-            col2.subheader('Predicted class: Healthy')
+            col2.markdown('<span class="healthy">Predicted class: Healthy</span>', unsafe_allow_html=True)