Update app.py

app.py

@@ -6,17 +6,22 @@ from sklearn.ensemble import IsolationForest
 from sklearn.preprocessing import StandardScaler
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 import torch
+from huggingface_hub import login
+import os
 
 st.set_page_config(page_title="Smart Factory RAG Assistant", layout="wide")
 
 st.title("🏠 Industry 5.0 | Smart Factory RAG Assistant (Open Source)")
 
-# Load the open-source model (Mistral-7B-Instruct)
+# Authenticate Hugging Face if using private token
+# login(os.getenv("HUGGING_FACE_HUB_TOKEN"))  # Uncomment if token is needed
+
+# Load open-access model (Zephyr)
 @st.cache_resource(show_spinner=True)
 def load_model():
-    tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.2")
+    tokenizer = AutoTokenizer.from_pretrained("HuggingFaceH4/zephyr-7b-beta")
     model = AutoModelForCausalLM.from_pretrained(
-        "mistralai/Mistral-7B-Instruct-v0.2",
+        "HuggingFaceH4/zephyr-7b-beta",
         torch_dtype=torch.float16,
         device_map="auto"
     )