Update app.py

app.py

@@ -40,7 +40,7 @@ add_sidebar_selectbox = st.sidebar.selectbox(
 
 def home():
     st.write('''
-        # Final Project AI Engineering
+        # Mineral Classification
         Hello, I'm Ahmad Alfian Faisal. This is the deployed result of my AI project, a deep learning model, which is part of my Final Project for the AI Engineering SkillAcademy at Ruangguru. If you have any questions or feedback, feel free to reach out to me via [LinkedIn](www.linkedin.com/in/ahmadalfianfaisal).
         ''')
     st.write('''
@@ -107,7 +107,7 @@ def prediction():
 
         elif model_name == "MobileNet":
             model_path = hf_hub_download(repo_id="ahmadalfian/mineral-classification",
-                                         filename="mobilenet_finetuned.pth")
+                                         filename="mobileNetV2_finetuned.pth")
             model = models.mobilenet_v2(pretrained=False)
             num_classes = 36
             model.classifier[1] = torch.nn.Linear(in_features=1280, out_features=num_classes)