Update app.py

app.py

@@ -8,9 +8,12 @@ from PIL import Image
 import uuid # For generating unique filenames
 from transformers import TFAutoModelForSequenceClassification
 from transformers import pipeline
+from huggingface_hub import hf_hub_download
+
+model_path = hf_hub_download(repo_id="methestrikerx100/Mineral_Identification_AI", filename="B7_MODEL_Final.h5")
+model = load_model(model_path)
 sentiment = pipeline("sentiment-analysis")
 # Load your pre-trained model
-model = load_model(r"C:\Users\nanom\OneDrive\Desktop\midterm implementation\B7_MODEL_Final.h5")
 
 # Function to preprocess the image for prediction
 def preprocess_image(img_array, s=224):